ZOJ Problem Set - 1038 T9

最新推荐文章于 2018-05-12 11:50:33 发布

原创最新推荐文章于 2018-05-12 11:50:33 发布 · 505 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#string #stdstring #null #struct

字典树专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种中等难度的字典树实现方法，通过DFS和优先队列结合使用来解决问题。文章详细展示了如何利用字典树进行字符串匹配，并通过实例代码说明了插入和搜索的操作过程。

中等难度的字典树。

用到的是DFS+优先队列的方法。比较容易实现，但是效率不太高~ 10ms。（cin,cout）；

首先，插入，每个节点要其记录权值。然后DFS每个点，push入队列，对<的重载也很容易写。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <algorithm>
#include <vector>
#include <limits.h>
#include <queue>
#include <stack>
using namespace std;
string a[11];
struct trie
{
    int p;
    trie*  next[26];
};
struct word
{
    string s;
    int pri;
};
priority_queue<word> q;
string que;
int len;
void inst(trie* root,string s,int pp)
{
    int i = 0;
    trie* p = root;
    int len = s.size();
    while(i < len)
    {
        int  k = s[i]-'a';
        if(p->next[k] == NULL)
        {
            trie* tmp = new trie;
            for(int j = 0;j < 26;j ++)
            tmp->next[j] = NULL;
            tmp->p = pp;
            p->next[k] = tmp;
        }
        else
        {
            p->next[k]->p += pp;
        }
        p = p->next[k];
        i ++;
    }
}

void dfs(int id,trie *r,string ss)
{
    int i,j,l;
    string sss;
    for(i = id;i < len-1&&i < id+1; i ++)
    {
        if(que[i]!='7'&&que[i]!='9')
        l = 3;
        else l = 4;
        for(j = 0;j < l;j ++)
        {
            int k = a[que[i]-'0'][j] - 'a';
            //cout<<k<<endl;
            if(r->next[k]!=NULL)
            {
                word tmp;
                sss = ss;
                sss += char(k +'a');
                //cout<<sss<<" "<<k<<endl;
                tmp.s = sss;
                tmp.pri = r->next[k]->p;
                q.push(tmp);
                dfs(i+1,r->next[k],sss);
            }
        }
    }
}
bool operator < (const word &a,const word &b)
{
    if(a.s.size() != b.s.size()) return a.s.size() > b.s.size();
    else if(a.pri!=b.pri) return a.pri < b.pri;
    else return a.s > b.s;
}
int main()
{
    a[2]="abc";a[3]="def";a[4]="ghi";
    a[5]="jkl";a[6]="mno";a[7]="pqrs";
    a[8]="tuv";a[9]="wxyz";
    int t,w,m,i;
    string str;
    cin>>t;
    trie *root = new trie;
    int tt = 0;
    while(t --)
    {
        for(i = 0;i < 26;i ++)
        root->next[i] = NULL;
        root->p = 0;
        cin>>w;
        for(i = 0;i < w;i ++)
        {
            cin>>str>>m;
            inst(root,str,m);
        }
        cin>>w;
        cout<<"Scenario #"<<++tt<<":"<<endl;
        for(i = 0 ;i < w;i ++)
        {
            cin>>que;
            len = que.size();
            dfs(0,root,"");
            for(int j = 0;j < len-1;j ++)
            {
                if(!q.empty()){
                cout<<q.top().s<<endl;
                while(!q.empty())
                {
                    if(q.top().s.size()==j+1) q.pop();
                    else break;
                }
                }
                else cout<<"MANUALLY"<<endl;
            }
            cout<<endl;
        }
        cout<<endl;
    }
}