题意:给出一个长度为 N N N的序列 q q q。
按顺序给出 M M M条信息,每条信息表示区间 [ A i , B i ] [A_i,B_i] [Ai,Bi]的和为 S i Si Si。
如果第 i i i条信息和之前已知的信息冲突,那么这条信息是无效的。
求一共有多少条无效信息。
( N , M ≤ 2 ∗ 1 0 5 ) (N,M\le 2*10^5) (N,M≤2∗105)
我们考虑对第 i i i条信息建立一个关系 A i → B i A_i \to B_i Ai→Bi,表示 ∑ j = A i B i q j \sum\limits_{j=A_i}^{B_i}q_j j=Ai∑Biqj即 S i S_i Si
这其实是前缀和的差值 S u m A i + S i = S u m B i Sum_{A_i}+S_i=Sum_{B_i} SumAi+Si=SumBi
某一个点 P P P可能有很多个点有关系。这些关系不需要一一表示出来
所有和 P P P有关系的点显然可以看作一个集合
就像一颗树,用一个点 X X X作为这个集合的根
那么我们只需要分别记录 X → P X \to P X→P也就是 X X X和 P P P的关系,就可以表示出 P P P现有的所有关系。
假设现在处理到 ( A i , B i , S i ) (A_i,B_i,S_i) (Ai,Bi,Si)。
V a l ( A i ) = F a t h e r ( A i ) → A i Val(A_i)=Father(A_i) \to A_i Val(Ai)=Father(Ai)→Ai和 V a l ( B i ) = F a t h e r ( B i ) → B i Val(B_i)=Father(B_i) \to B_i Val(Bi)=Father(Bi)→Bi
如果 A i A_i Ai和 B i B_i Bi本来就间接有关系了,那么 F = F a t h e r ( A i ) = F a t h e r ( B i ) F=Father(A_i)=Father(B_i) F=Father(Ai)=Father(Bi)
可以利用 V a l ( A i ) = F → A i Val(A_i)=F \to A_i Val(Ai)=F→Ai和 V a l ( B i ) = F → B i Val(B_i)=F \to B_i Val(Bi)=F→Bi来得到 A i A_i Ai和 B i B_i Bi已知的关系
即 A i → B i = V a l ( B i ) − V a l ( A i ) A_i \to B_i=Val(B_i)-Val(A_i) Ai→Bi=Val(Bi)−Val(Ai)
显然,这条信息是否合法可以根据 V a l ( B i ) − V a l ( A i ) Val(B_i)-Val(A_i) Val(Bi)−Val(Ai)是否等于 S i S_i Si来判定。
如果 A i A_i Ai和 B i B_i Bi本来没有关系,现在它们有关系了, F a t h e r ( A i ) Father(A_i) Father(Ai)和 F a t h e r ( B i ) Father(B_i) Father(Bi)也有关系了
我们就要把它们关联起来,比方说建立 F a t h e r ( B i ) → F a t h e r ( A i ) Father(B_i) \to Father(A_i) Father(Bi)→Father(Ai)
更新 F a t h e r ( F a t h e r ( A i ) ) = F a t h e r ( B i ) Father(Father(A_i))=Father(B_i) Father(Father(Ai))=Father(Bi)
V a l ( F a t h e r ( A i ) ) = V a l ( B i ) + S i − V a l ( A i ) Val(Father(A_i))=Val(B_i)+S_i-Val(A_i) Val(Father(Ai))=Val(Bi)+Si−Val(Ai)
以上,建一个带权并查集就可以解决这道题
值得一提的是本题的区间实际上是离散的点,所以区间要建成左开右闭或者左闭右开
可能这么讲比较容易懂: A i → B i = S u m [ B i ] − S u m [ A i − 1 ] A_i \to B_i = Sum[B_i] - Sum[A_i-1] Ai→Bi=Sum[Bi]−Sum[Ai−1]
类似的例子在线段树题里面也有,一个经典的数据就是 [ 1 , 5 ] [1,5] [1,5]和 [ 6 , 10 ] [6,10] [6,10]这两个区间
注意路径压缩的时候也要更新权值
#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<algorithm>
#include<cmath>
using namespace std;
int N,M;
int fa[200005]={},val[200005]={};
int find(int x)
{
if(x==fa[x])return x;
else
{
int top=find(fa[x]); //注意一定要先find,合并上面的权值到fa[x]
val[x]+=val[fa[x]];
return fa[x]=top;
}
}
int main()
{
while(scanf("%d%d",&N,&M)!=EOF)
{
for(int i=0;i<=N;++i)fa[i]=i,val[i]=0;
int Ans=0;
for(int ta,tb,tc,ffa,ffb,i=1;i<=M;++i)
{
scanf("%d%d%d",&ta,&tb,&tc);
--ta;
ffa=find(ta),ffb=find(tb);
if(ffa!=ffb)
{
fa[ffb]=ffa;
val[ffb]=val[ta]-val[tb]+tc;
}
else if(val[tb]-val[ta]!=tc)++Ans;
}
printf("%d\n",Ans);
}
return 0;
}