poj 2001 Shortest Prefixes(字典树)

来源：互联网发布：喵喵折for mac 编辑：程序博客网时间：2024/05/01 18:38

题目链接：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]++;    }}

1 0