洛谷 P2342 叠积木 题解

本蒟蒻又来发题解了

这题是不是有点像并查集,但是那个询问的个数是不是有点骚;

所以,普通的并查集是无法解决这个问题的,这个时候就需要用到带权并查集了;

每次跑的时候都记录下它的下面有几个点,然后询问的时候的复杂度是O(1)的;

好,接下来就可以愉快的码代码了:

#include<bits/stdc++.h>
using namespace std; #define maxn 500010 char ch; int n, x, y, fa[maxn], sum[maxn], num[maxn]; //比常规的并查集多了一行 int find(int x) { if(fa[x] == x) return x; sum[x] += sum[fa[x]];//还需要统计它下面的积木数 return fa[x] = find(fa[x]); } //fa[]就不说了,sum[]是来存它下面的积木数,num[]是这一列的积木数; int main() { scanf("%d", &n); for(int i = 1; i <= maxn; ++ i) { fa[i] = i;//嗯,它的祖先就是它自己,常规操作; sum[i] = 0;//它的下面QwQ肯定没有积木啊 num[i] = 1;//目前这一列只有一个积木 } for(int i = 1; i <= n; ++ i) { cin >> ch; if(ch == 'M')//如果这是'M',那就要合并 { scanf("%d%d", &x, &y); x = find(x), y = find(y); if(x != y)//如果他们的祖先不同,就是他们不在同一个联通块上 { fa[x] = y;//合并它们 sum[x] = num[y]; //数量要相加 num[y] += num[x]; //因为x要和y合并 } } else { scanf("%d", &x); find(x);//还需要再来一次; printf("%d\n", sum[x]);//O(1)输出; } } return 0; }

嗯,你愉快的拿了27分;

好,开始查错了,

戴上眼镜,睁大眼睛

哪里有错?

就是过不去数据,哎;

好吧,我猜这么神仙的你们肯定找到哪里错了,不像蒟蒻我(划掉

#include<bits/stdc++.h>
using namespace std; #define maxn 500010 char ch; int n, x, y, fa[maxn], sum[maxn], num[maxn]; //fa[]就不说了,sum[]是来存它下面的积木数,num[]是这一列的积木数; //比常规的并查集多了一行 int find(int x) { if(fa[x] == x) return x; int f = find(fa[x]); sum[x] += sum[fa[x]];//还需要统计它下面的积木数 return fa[x] = f; } int main() { scanf("%d", &n); for(int i = 1; i <= maxn; ++ i) { fa[i] = i;//嗯,它的祖先就是它自己,常规操作; sum[i] = 0;//它的下面QwQ肯定没有积木啊 num[i] = 1;//目前这一列只有一个积木 } for(int i = 1; i <= n; ++ i) { cin >> ch; if(ch == 'M')//如果这是'M',那就要合并 { scanf("%d%d", &x, &y); x = find(x), y = find(y); if(x != y)//如果他们的祖先不同,就是他们不在同一个联通块上 { fa[x] = y;//合并它们 sum[x] = num[y]; //数量要相加 num[y] += num[x]; //因为x要和y合并 } } else { scanf("%d", &x); find(x);//还需要再来一次; printf("%d\n", sum[x]);//O(1)输出; } } return 0; }

有没有发现哪里错了,

是的,就是在find函数里面;

第一个代码是:

int find(int x) { if(fa[x] == x) return x; sum[x] += sum[fa[x]];//还需要统计它下面的积木数 return fa[x] = find(fa[x]); }

第二个代码是:

int find(int x) { if(fa[x] == x) return x; int f = find(fa[x]); sum[x] += sum[fa[x]];//还需要统计它下面的积木数 return fa[x] = f; } 

是吧,第一个是先更新sum[],再递归;

而第二个是先递归,再更新sum[];

所以,我们应该是要先递归完,再来更新;

如果要先更新的话,那这个find函数对与sum[]的更新就没有作用;

嗯,就这么结束了;

PS. AC的代码是第二篇,第一篇有问题;

猜你喜欢

转载自www.cnblogs.com/Flash-plus/p/11210692.html