POJ2001Shortest Prefixes(字典树)

题目大意就是帮你给N条字符串，每条长度不超过20。问要将他们单一识别出来，每个字符串最短可以缩为多短。

如：

abc

abcdefg

bc

adef

这四个字符串就可分别缩写为

abc

abcd

b

ad

方法：   字典树（可以参阅http://s.acmore.net/show_article/show/58）。

另外我还用了一个bool数组last用来记录每个单一识别的字符串最短可以到哪个位置,他的下标就是字典树中每个字母对应的序号

方法如下：（以上面的为例）

当输入的字符串在某一个位置开始与之前的不同，就记录这个不同的字母（设置为true），之后的不再改变

当输入字符串的某位置在建好的树中时，把last加长（设置为true）

第一次输入：abc          last[0]=true;

第二次输入：abcdefg    last[0]=last[1]=last[2]=last[3]=true;

第三次输入：bc            last[0]=true;

第四次输入：adef         last[0]=last[1]=true;

这样一来输出端长度就分别为1、4、1、2

代码实现如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <stack>
#include <set>
#include <queue>
#define MAX(a,b) (a) > (b)? (a):(b)
#define MIN(a,b) (a) < (b)? (a):(b)
#define mem(a) memset(a,0,sizeof(a))
#define INF 1000000007
#define MAXN 20005
using namespace std;

int trie[MAXN][26],val[MAXN],S;
bool last[MAXN];
char ma[1005][30];

int get_num(char ch){return ch-'a';}

void init()
{
    mem(trie);mem(val);
    mem(last);mem(ma);
    S = 1;
}

void insert(char *s)
{
    int u=0,len = strlen(s);
    int flag = 1;
    for(int i=0;i<len;i++)
    {
        int c = get_num(s[i]);
        if(!trie[u][c])
        {
            trie[u][c] = S;
            val[u] = 1;
            if(flag)//如果与上面的字符串开始不一样，酒吧第一个开始不一样的位置设置为true
            {
                last[u]=true;
                flag = 0;//之后的依然是false
            }
            u = S;
            S++;
        }
        else if(val[u] == 1)
        {
            u = trie[u][c];
            last[u] = true;//在查找中发现与已建好的树里面相同的数，那么需要识别的字符串就要相应的加长
        }
    }
    if(val[u]==1 || val[u] == 2)return;
    else val[u] = 2;
}

int main()
{
    init();
    int T =0;
    while(scanf("%s",ma[T])!=EOF)
    {
        insert(ma[T++]);
    }
    for(int i=0;i<T;i++)
    {
        printf("%s ",ma[i]);
        int u=0,j=0;
        while(last[u] && ma[i][j])
        {
            printf("%c",ma[i][j]);
            u = trie[u][get_num(ma[i][j++])];
        }
        printf("
");
    }
    return 0;
}