poj 2001 Shortest Prefixes(字典树)

题目链接：poj 2001 Shortest Prefixes

题目大意：给定一个字符串集合，要求使得每个字符串尽量缩短，但是仍然能区分开。

解题思路：根据字符串集建立字典树，并且每插入一个字符串，所有路径上节点均加1，表示该子串是这个字符串的前

缀。然后对每个字符串进行一次查找，当碰到节点的统计值为1时，表示当前子串仅为该字符串的前缀，即为要找的字

符串。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 20005;
const int sigma_size = 26;

struct Tire {
    int sz;
    int g[maxn][sigma_size];
    int val[maxn];

    void init();
    int idx(char ch);
    void insert(char* s);
    void find(char* s);
}T;

int N;
char w[1005][25];

int main () {
    N = 0;
    T.init();

    while (scanf("%s", w[N]) == 1)
        T.insert(w[N++]);

    for (int i = 0; i < N; i++) {
        printf("%s ", w[i]);
        T.find(w[i]);
        printf("\n");
    }
    return 0;
}

void Tire::init() {
    sz = 1;
    val[0] = 0;
    memset(g[0], 0, sizeof(g[0]));
}

int Tire::idx (char ch) {
    return ch - 'a';
}

void Tire::find(char* s) {
    int u = 0, n = strlen(s);

    for (int i = 0; i < n; i++) {
        int v = idx(s[i]);

        u = g[u][v];
        printf("%c", s[i]);
        if (val[u] == 1)
            return;
    }
}

void Tire::insert(char* s) {
    int u = 0, n = strlen(s);

    for (int i = 0; i < n; i++) {
        int v = idx(s[i]);

        if (g[u][v] == 0) {
            val[sz] = 0;
            memset(g[sz], 0, sizeof(g[sz]));
            g[u][v] = sz++;
        }

        u = g[u][v];
        val[u]++;
    }
}