【D - How Many Answers Are Wrong】

思路:

  • 带权并查集,区间统计

  • 带权并查集普通并查集的区别:

    1. Find 函数不可以写递归了,要能够动态更新路径上所有点的 val
    2. UNION 函数中要改 val[parr],不需要改val[r]
  • 本题注意:

    1. 题目没说多组用例,自己也要读到文件尾,不然就会WA
    2. 区间统计需要变成左开右闭(如此针对: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;
}

猜你喜欢

转载自blog.csdn.net/flash403/article/details/94330063