hdu3038(并查集)

题目链接

题意:有n次询问,给出a到b区间的总和,问这n次给出的总和中有几次是和前面已近给出的是矛盾的

分析:sum数组维护着到根节点的距离(即区间和),每次合并x,y,s(a,b分别为x,y的根节点)时(假设由a指向b),可根据sum[x](x到达a的距离),sum[y](y到达b的距离),s(x到达y的距离),然后推出x的根节点a到达y的根据点b的距离,剩下x到达y的根节点a的距离,再在find函数里再更新。

#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <queue>
#include <cstdlib>
#include <stack>
#include <vector>
#include <set>
#include <map>
#define LL long long
#define mod 100000000
#define inf 0x3f3f3f3f
#define eps 1e-9
#define N 200010
#define FILL(a,b) (memset(a,b,sizeof(a)))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define PII pair<int,int>
using namespace std;
int fa[N],vis[N],sum[N];
int find(int x)
{
if(x==fa[x])return x;
int pa=fa[x];
fa[x]=find(fa[x]);
//当父节点pa指向新的根节点时,sum[x]是x到达父节点pa的距离,pa到达的新根节点之前已更新好
//因此sum[x]+sum[pa]就是x到达新根节点的距离了
sum[x]=sum[x]+sum[pa];
return fa[x];
}
void merge(int x,int y,int s)
{
int a=find(x);
int b=find(y);
//sum[y]表示y到b的距离,sum[x]表示x到a的距离,s表示x到y的距离,模拟一下得出a到b的距离或b到a的距离
if(a>b)
{
fa[b]=a;
sum[b]=sum[x]-s-sum[y];
}
else
{
fa[a]=b;
sum[a]=sum[y]+s-sum[x];
}
}
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)>)
{
for(int i=;i<=n;i++)
{
fa[i]=i;sum[i]=;
}
int ans=;
for(int i=;i<=m;i++)
{
int x,y,s;
scanf("%d%d%d",&x,&y,&s);
x--;
if(find(x)==find(y))
{
if(sum[x]!=sum[y]+s)
ans++;
}
else merge(x,y,s);
}
printf("%d\n",ans);
}
}
上一篇:7.java.lang.IllegalAccessException


下一篇:Struts2 实现文件上传