POJ 2513 Colored Sticks 【Trie树】

Trie树+欧拉路径的判定。

并查集来判断连通性，然后判断欧拉路径的条件就可以了。

Trie树的插入操作和查询Index操作，其他操作没有写。

没有写全的模板。

#include <cstdio>
#include <cstring>
using namespace std;
#define N 500002
struct node {
    int num;
    bool is_word;
    node *next[26];
    node() {
        num = 0;
        is_word = false;
        memset(next, NULL, sizeof(next));
    }
} root;
int n, c[N] = {0}, f[N] = {0};

void Insert(char s[]) {
    node *p = &root;
    for (int i=0; s[i]; i++) {
        int x = s[i] - 'a';
        if (p->next[x] == NULL)
            p->next[x] = new node;
        p = p->next[x];
    }
    if (!p->is_word) {
        p->is_word = true;
        p->num = ++n;
    }
}
int Index(char s[]) {
    node *p = &root;
    for (int i=0; s[i]; p=p->next[s[i]-'a'], i++) ;
    return p->num;
}
int find_set(int x) {
    if (x == f[x]) return x;
    return f[x] = find_set(f[x]);
}
int main() {
    char a[11], b[11];
    n = 0;
    int ia, ib;
    while (scanf(" %s %s", a, b) == 2) {
        Insert(a); Insert(b);
        ia = Index(a); ib = Index(b);
        c[ia]++, c[ib]++;
        if (f[ia] == 0) f[ia] = ia;
        if (f[ib] == 0) f[ib] = ib;
        f[find_set(ia)] = f[find_set(ib)];
    }
    int cnt = 0;
    bool flag = true;
    for (int i=1; i<=n; i++) {
        if (c[i] & 1) cnt++;
        if (find_set(i) != f[1])
            flag = false;
    }
    if ((cnt==0 || cnt==2) && flag) printf("Possible\n");
    else printf("Impossible\n");


    return 0;
}