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
*/