最近一連做了好幾道就着區間建圖的圖論題,發現建圖的時候都很喜歡把區間看成兩個前綴和之差,我小叮噹長見識了。這道題也是這樣。一開始的時候想讓r數組表示當前點到其祖先這個區間(包含兩端點),後來要路徑壓縮了,發現情況不大對頭。。。前綴和大法好!果然應該記錄當前點與祖先的前綴和之差!
#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
using namespace std;
int n,m,fa[200010],r[200010];
int find(int x)
{
int fx=fa[x];
if (fa[x]!=x)
{
fx=find(fa[x]);
r[x]+=r[fa[x]];
}
return fa[x]=fx;
}
bool u(int x,int y,int v)
{
int fx=find(x);
int fy=find(y);
if (fx==fy)
{
if (r[y]-r[x]==v) return true;
else return false;
}
fa[fy]=fx;
r[fy]=r[x]+v-r[y];
return true;
}
int main()
{
while (~scanf("%d%d",&n,&m))
{
for (int i=0; i<=n; i++)
{
fa[i]=i;
r[i]=0;
}
int ans=0;
for (int i=1; i<=m; i++)
{
int a,b,s;
scanf("%d%d%d",&a,&b,&s);
a--;
if (!u(a,b,s)) ans++;
}
cout<<ans<<endl;
}
return 0;
}