POJ 2001 Shortest Prefixes

传送门

https://vjudge.net/problem/POJ-2001

题目大意就是求每一个字符串的最短前缀，并且这个前缀其他字符串都没有。

一道trie树板子题。建树的时候，每一个结点维护的是经过这个状态的字符串有多少个。然后查询的时候，对于每一个字符串，从头开始跑trie树，找到第一个数是1的结点。那么从头到这个节点经过的边就是所求前缀。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iostream> 
using namespace std;
typedef long long ll;
#define enter printf("
")
const int maxn_num = 2e4 +5;
const int maxn_size = 30;
const int INF = 0x3f3f3f3f;

struct Trie
{
    int ch[maxn_num][maxn_size], val[maxn_num];        
    /*ch[i][j]代表第i个结点，经过边权为j的边所到达的结点 
    所以ch的第一维应该开字符串数目*字符种类数*/
    int tot;        //结点个数                        
    Trie()
    {
        tot = 1; memset(val, 0, sizeof(val));
        memset(ch, 0, sizeof(ch));
    }
    int change(char c) {return c - 'a';} 
    void Insert(char *s)
    {
        int now = 0;
        for(int i = 0; i < strlen(s); ++i)
        {
            int c = change(s[i]);
            if(!ch[now][c]) ch[now][c] = tot++;        //新开结点 
            now = ch[now][c]; val[now]++;
        }
    }
    void query(char *s)
    {
        int now = 0;
        for(int i = 0; i < strlen(s); ++i)
        {
            printf("%c", s[i]);
            int c = change(s[i]);
            if(val[ch[now][c]] == 1) break;
            now = ch[now][c];
        }
        enter;
    }
}trie;

char s[1005][50];

int main()
{
    int x = 0;
    while(scanf("%s", s[++x]) != EOF)
        trie.Insert(s[x]);
    for(int i = 1; i <= x; ++i) 
    {
        printf("%s ", s[i]);
        trie.query(s[i]);
    }
    return 0;
}