入门解释:http://www.cnblogs.com/xzxl/p/7226557.html
进阶解释:http://www.cnblogs.com/xzxl/p/7341536.html
入门篇模板
#define ARR_SIZE 100
int pre[ARR_SIZE];
int deep[ARR_SIZE];
void init() {
int i = 0;
for (i = 0; i < ARR_SIZE; i++) {
pre[i] = i;
deep[i] = 1;
}
}
int pre_find(int x) {
if (pre[x] == x)
{
return x;
}
return pre_find(pre[x]); // 递归查找
}
void unite(int x, int y) {
int i = pre_find(x);
int j = pre_find(y);
if (i == j){
return;
}
if (deep[i] > deep[j]) {
pre[j] = i;
}
else {
if (deep[i] == deep[j]) {
deep[j]++;
}
pre[i] = j;
}
}
进阶篇解题
食物链
动物王国中有三类动物A,B,C,这三类动物的食物链构成了有趣的环形。A吃B, B吃C,C吃A。
现有N个动物,以1-N编号。每个动物都是A,B,C中的一种,但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述:
第一种说法是"1 X Y",表示X和Y是同类。
第二种说法是"2 X Y",表示X吃Y。
此人对N个动物,用上述两种说法,一句接一句地说出K句话,这K句话有的是真的,有的是假的。当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
1) 当前的话与前面的某些真的话冲突,就是假话;
2) 当前的话中X或Y比N大,就是假话;
3) 当前的话表示X吃X,就是假话。
你的任务是根据给定的N(1 <= N <= 50,000)和K句话(0 <= K <= 100,000),输出假话的总数。
Input
第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
Output
只有一个整数,表示假话的数目。
Sample Input
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
Sample Output
3
#define ARR_SIZE 100
int pre[ARR_SIZE]; // 存父亲节点
int r[ARR_SIZE]; // 存与根节点的关系
// r[x] = 0 : x 与根节点同类
// r[x] = 1 : x 被根节点吃
// r[x] = 2 : x 吃根节点
int n, k;
void init() {
int i = 0;
for (i = 0; i < n; i++) {
pre[i] = i;
r[i] = 0; // 与自身同一种类
}
}
int find(int x) {
if (pre[x] == x) {
return x;
}
int tmp = pre[x];
pre[x] = find(pre[x]); // 增加路径压缩算法
r[x] = (r[x] + r[tmp]) % 3; //回溯由子节点与父节点的关系和父节点与根节点的关系找子节点与根节点的关系 %3
return pre[x];
}
void unite(int x, int y, int d) {
int fx = find(x);
int fy = find(y);
if (fx == fy) return;
pre[fy] = fx; //合并树 注意:被x吃,所以以 x的根为父
r[fy] = (r[x] - r[y] + d - 1 + 3) % 3; // 对应更新与父节点的关系
}
int main() {
printf("input n :\n");
scanf("%d",&n);
printf("input k :\n");
scanf("%d",&k);
init();
int sum = 0; // 假话数量
int D, a, b;
while (k--) {
printf("input D a b :\n");
scanf("%d%d%d", &D, &a, &b);
if (a > n || b > n || (D == 2 && a == b)) {
sum++;
continue;
}
else if (find(a) == find(b)) {
if (D == 1 && r[a] != r[b]) // 同类但是关系不同
sum++;
if (D == 2 && (r[a] + 1) % 3 != r[b]) // 如果a没有吃b (注意要对应unite(a,b)的情况,否则一路WA到死啊!!!)
sum++;
}
else {
unite(a, b, D);
}
}
printf("sum is [%d]\n", sum);
return 0;
}