[POJ2513]Colored Sticks(Trie+并查集)

Colored Sticks

题目大意

有不多于250000根木棒,每根木棒的两端都有一个长度不大于10的字符串代表它们的颜色,如果两根木棒的分别一端是同一种颜色,就可以把这两根木棒连接在一起,问是否可能把所有木棒连在一起。

解法

第一反应是开一棵Trie,然后把所有的颜色都扔在里面,一边扔一边计数,并且给每种颜色的最后一个字母打上标记。显然,在所有打上标记的节点中,没有或者有且仅有两个出现次数为奇数的节点,才能构成满足题意的情况。

因为我太菜了,然后我就写了个裸的Trie交了一发WA。调了半天才想起来没有判断颜色的连通性。于是用每种颜色的最后一个字母的下标代表当前的颜色,丢到并查集里,最后检查一下连通性就可以了。

我太菜了啊。。我什么都不会啊。。

贴代码,写的很丑,你们不要D我>_<

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#include <iostream>
#include <sstream>
#include <iomanip>
#include <algorithm>
#include <map>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstring>
#include <cmath>
#include <cstdio>
#define pii pair<int, int>
#define makep make_pair
#define pb push_back
#define fi first
#define se second
using namespace std;

typedef long long ll;
typedef unsigned long long ull;

int main0 ();
int main () {
#ifndef ONLINE_JUDGE
freopen("C:\\Users\\98497\\Desktop\\code\\file.in", "r", stdin);
#endif
ios::sync_with_stdio(false);
main0();
#ifndef ONLINE_JUDGE
fclose(stdin);
#endif
return 0;
}

const int INF = 0x3f3f3f3f;
const int N = 1e6;

int tr[N][26];
int cnt[N];
bool vis[N];
int p = 0;

int fa[N];

int getfa (int a) {
if (fa[a] == a)
return a;
else
return fa[a] = getfa(fa[a]);
}

void merge (int a, int b) {
a = getfa(a);
b = getfa(b);
fa[a] = b;
}

void insert (char s1[], char s2[]) {
int rt1 = 0;
for (int i = 0;i < strlen(s1);i++) {
int id = s1[i] - 'a';
if (!tr[rt1][id]) tr[rt1][id] = ++p;
cnt[tr[rt1][id]]++;
rt1 = tr[rt1][id];
}
vis[rt1] = true;

int rt2 = 0;
for (int i = 0;i < strlen(s2);i++) {
int id = s2[i] - 'a';
if (!tr[rt2][id]) tr[rt2][id] = ++p;
cnt[tr[rt2][id]]++;
rt2 = tr[rt2][id];
}
vis[rt2] = true;

merge(rt1, rt2);
}

int main0 () {
memset(cnt, 0, sizeof cnt);
memset(vis, 0, sizeof vis);
for (int i = 0;i < N;i++) fa[i] = i;

char s1[20];
char s2[20];
while (scanf("%s%s", s1, s2) == 2)
insert(s1, s2);

bool flag = true;

int odd = 0;
int cmp = 0;
for (int i = 0;i <= p;i++) if (vis[i]) {//cout << cnt[i]<<' '<<cnt2[i]<<endl;
if (cnt[i] % 2)
odd++;

if (cmp == 0)
cmp = i;
else
if (getfa(i) != getfa(cmp)) flag = false;
}
if ((odd != 0) && (odd != 2))
flag = false;
printf(flag ? "Possible" : "Impossible");
}