10.16考试后

T2
题意
有一个长度为𝑛的数列,数列中每个数都是[0, 𝑝−1]之间的整数。小A不知道数列中
每个数的值,所以向小B做了𝑚次询问。每次小A会向小B询问一个区间[𝑙, 𝑟]中所有
数的和对𝑝取模的结果。问完所有问题后,小A发现小B的回答中似乎存在矛盾。现在
小A想找到最大的𝑋,满足小B的前𝑋次回答中不存在矛盾(𝑋有可能等于𝑚)。
思路:可以用边带权并查集维护关系;
维护与根节点的差值,注意一下合并就好了

#include<iostream>
#include<cstdio>
using namespace std;
const int N=1e6+7;
int n,m,p,ans,flag=0;
int fa[N],d[N];
int find(int x){
	int tt;
	if(x==fa[x]) return x;
	tt=find(fa[x]);
	d[x]+=d[fa[x]];
	return fa[x]=tt;
}
int main(){
	freopen("seq.in","r",stdin);
	freopen("seq.out","w",stdout);
	scanf("%d%d%d",&n,&m,&p);
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=m;i++){
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		if(flag==1) continue;
		int tx=find(x-1);
		int ty=find(y);
		if(tx==ty){
			if((d[y]-d[x-1]+p)%p!=z){
				cout<<i-1<<"\n";
				flag=1;
			}
		}else{
			fa[tx]=ty;
			d[tx]=z-d[y]-d[x-1];
		}
	}
	if(flag==0) cout<<m;
	fclose(stdin);
	fclose(stdout);
	return 0;
}
/*
10 5 2
1 2 0
3 4 1
5 6 0
1 6 0
7 10 1
*/
posted @ 2020-10-16 10:35  Aswert  阅读(131)  评论(0编辑  收藏  举报