Trie树（SOJ3336）

(2012-07-19 18:46:00)

SOJ3336: http://cstest.scu.edu.cn/soj/problem.action?id=3336

这道题在分类中写的是用Trie树。刚开始我用KMP做的，结果超时了。后来看到有人用26进制+map做的，试了一下过了。

其实所谓的26进制，只是将字符串转化成一个long long型，对应的map由原来的map<string,int>改成map<long long,int>。具体来讲就是：将'a'作为0，将'b'作为1,...,将'z'作为25，将一个具体的字符串转换成long long型。

这道题我是将n个code用map存起来，然后将text每10个字符作26进制处理，再在map中查找。

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#include<map>
using namespace std;
char text[1000005];
struct node
{
    char code[12];
    char name[12];
}s[100005];
map<long long,int>xiaoye;
long long exchange(int i)
{
    long long result=0;
    result=text[i]-'a';
    result=result*26+text[i+1]-'a';
    result=result*26+text[i+2]-'a';
    result=result*26+text[i+3]-'a';
    result=result*26+text[i+4]-'a';
    result=result*26+text[i+5]-'a';
    result=result*26+text[i+6]-'a';
    result=result*26+text[i+7]-'a';
    result=result*26+text[i+8]-'a';
    result=result*26+text[i+9]-'a';
    return result;
}
int tt[100005];
int main()
{
    int test;
    int n;
    int i;
    scanf("%d",&test);
    while(test--)
    {
        scanf("%s",text);
        scanf("%d",&n);
        for(i=0;i<n;i++)
        {
            scanf("%s%s",s[i].name,s[i].code);
            long long result=0;
            result=s[i].code[0]-'a';
            result=result*26+s[i].code[1]-'a';
            result=result*26+s[i].code[2]-'a';
            result=result*26+s[i].code[3]-'a';
            result=result*26+s[i].code[4]-'a';
            result=result*26+s[i].code[5]-'a';
            result=result*26+s[i].code[6]-'a';
            result=result*26+s[i].code[7]-'a';
            result=result*26+s[i].code[8]-'a';
            result=result*26+s[i].code[9]-'a';
            xiaoye[result]=i+1;
        }
        memset(tt,0,sizeof(tt));
        int len=strlen(text);
        for(i=0;i<=len-10;i++)
        {
            int temp=xiaoye[exchange(i)];
            if(temp)
                tt[temp-1]++;
        }
        int max=-1;
        int index;
        for(i=0;i<n;i++)
            if(tt[i]>max)
            {
                max=tt[i];
                index=i;
            }
        sort(tt,tt+n);
        if(tt[n-1]==tt[n-2])
            printf("No-solution!\n");
        else
            printf("%s\n",s[index].name);
        xiaoye.clear();
    }
    return 0;
}

下面说一下这道题的Trie树做法。这也是我过的第一道Trie树题。将text每10个字符作为一个单词建立Trie树，其中每个节点node含有一个属性num用于记录重复单词的个数。然后在Trie树种搜索n个code。

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
#define kind 26
using namespace std;
struct node
{
    int num;
    node *son[kind];
    node()
    {
        num=0;
        memset(son,NULL,sizeof(son));
    }
};
char text[1000005];
void insert(node *root,int index)
{
    int i;
    for(i=index;i<index+10;i++)
    {
        if(root->son[text[i]-'a'])
        {
            root=root->son[text[i]-'a'];
            if(i==index+9)
                root->num++;
            continue;
        }
        node *temp=new node();
        root->son[text[i]-'a']=temp;
        root=temp;
        if(i==index+9)
            root->num++;
    }
}
int search(node *root,char *pattern)
{
    int i;
    for(i=0;i<strlen(pattern);i++)
    {
        if(root==NULL)
            break;
        root=root->son[pattern[i]-'a'];
    }
    if(root)
        return root->num;
    else
        return 0;
}
void del(node *root)
{
    int i;
    for(i=0;i<kind;i++)
    {
        if(root->son[i])
            del(root->son[i]);
    }
    delete root;
}
struct hi
{
    char name[12];
    char code[12];
}s[100005];
int result[100005];
int main()
{
    int test;
    int n;
    int i;
    scanf("%d",&test);
    while(test--)
    {
        node *root=new node();
        
        scanf("%s",text);
        int len=strlen(text);
        for(i=0;i<=len-10;i++)
            insert(root,i);
        memset(result,0,sizeof(result));

        scanf("%d",&n);
        int index;
        int max=-1;
        for(i=0;i<n;i++)
        {
            scanf("%s",s[i].name);
            scanf("%s",s[i].code);
            result[i]=search(root,s[i].code);
            if(result[i]>max)
            {
                max=result[i];
                index=i;
            }
        }
        sort(result,result+n);
        if(result[n-1]==result[n-2])
            printf("No-solution!\n");
        else
            printf("%s\n",s[index].name);
        del(root);
    }
    return 0;
}