思路:
-
带权并查集,区间统计。
-
带权并查集和普通并查集的区别:
- Find 函数不可以写递归了,要能够动态更新路径上所有点的 val 。
- UNION 函数中要改 val[parr],不需要改val[r]。
-
本题注意:
- 题目没说多组用例,自己也要读到文件尾,不然就会WA。
- 区间统计需要变成左开右闭(如此针对:6,6,1 类限制才不会出错)。
代码:
- 46ms 2972kB
//46ms 2972kB
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 200005;
int N,M;
int ans;
int par[maxn];
int val[maxn];//value后缀和
void INIT(){
memset(par , -1 , sizeof(par));
memset(val , 0 , sizeof(val));
ans = 0;
return ;
}
int FIND(int i){
if(par[i] == -1)
return i;
int temp = par[i];
par[i] = FIND(par[i]);
val[i] += val[temp];
return par[i];
}
int main(){
while(cin>>N>>M){
INIT();
while(M--){
int l , r , w;
scanf("%d%d%d" , &l , &r , &w);
int parl = FIND(l-1) ;
int parr = FIND(r) ;
if(parr == parl)
if(val[l-1] + w != val[r])
ans++;
else ;
else{
par[parr] = parl;
val[parr] = val[l-1] + w - val[r] ;
//为什么不需要改 val[r] : par[r]依然是parr , 且parr和 r之间的势差仍为val[r]
}
}
printf("%d\n" , ans);
}
return 0;
}