PKU 2513 Colored Sticks

 trie树+并查集+欧拉回路

 

用trie树将字符串一一映射（貌似用hash也行）， 用并查集判断图是否联通，然后判断是否存在欧拉回路（不要以为回路不符合题意，其实只要拉直就好了），即度为奇数的点有两个或没有。

 

1. #include <stdio.h>
2. const int N = 250000 + 5;
3. int fa[N*2]={0};
4. int num;
6. struct Node{
7.     Node *branch[26];
8.     int key;
9.     Node(){
10.         for ( int i=0; i<26; i++ )
11.             branch[i] = NULL;
12.         key = 0;
13.     }
14. } *root;
15. void NewNode( Node *point, int cnum )
16. {
17.     Node *nnode = new Node();
18.     point->branch[cnum] = nnode;
19. }
21. int insert( char *word )
22. {
23.     Node *point = root;
24.     while ( *word!='/0' )
25.     {
26.         int cnum = *word-'a';
27.         if ( point->branch[cnum] == NULL  )
28.             NewNode( point , cnum );
29.         point = point->branch[cnum];
30.         word ++;
31.     }
32.     if ( point->key==0 )
33.     {
34.         point->key = ++num;
35.         fa[num-1] = num-1;
36.     }
37.     return point->key;
38. }
40. int v[N*2] = {0};
42. int Root( int x)
43. {
44.     int r = fa[x];
45.     if ( r!=x )
46.         fa[x] = Root( r );
47.     return fa[x];
48. }
50. int main ()
51. {
52.     num = 0;
53.     root = new Node();
54.     int t1, t2; 
55.     char s1[15], s2[15];
56.     while ( scanf("%s %s", s1, s2 )!=EOF )
57.     {
58.         t1 = insert( s1 );  
59.         t2 = insert( s2 );
60.         int ra = Root( t1-1 );
61.         int rb = Root( t2-1 );
62.         if ( ra!=rb )
63.             fa[rb] = ra;
64.         v[t1-1]++;
65.         v[t2-1]++;
66.     }
67.     int nodd = 0;
68.     int rr = Root( 0 );
69.     int flag = 1;
70.     for ( int i=0; i<num; i++ )
71.     {
72.         if ( v[i]%2==1 )
73.             nodd ++;
74.         if ( Root(i)!=rr )
75.         {
76.             flag = 0;
77.             break;
78.         }
79.     }
80.     if ( ( nodd==0 || nodd==2 ) && flag )
81.         printf("Possible/n");
82.     else
83.         printf("Impossible/n");
84.     return 0;
85. }