并查集练习---poj 1984
usaco的月赛题。
记录两个点之间x方向和y方向的相对距离,用并查集维护。
若与poj 1182食物链进行比较,便会发现路径压缩部分,集合合并部分的相似点。
所以并查集不难,是有一定套路可循的。大家一定要好好总结。
【代码】
[cpp]
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=40005,K=10005;
struct node
{
int x,y,idx,n;
}a[K];
int x[N],y[N],dx[N],dy[N],rx[N],ry[N],ans[K],fa[N];
int n,m,k;
int find(int x)
{
if (fa[x]==x) return x;
int t=fa[x];
fa[x]=find(fa[x]);
rx[x]+=rx[t];
ry[x]+=ry[t];
return fa[x];
}
int cmp(node a,node b)
{
return a.idx<b.idx;
}
int main()
{
int i,j,d,fx,fy;
char c;
freopen("in","r",stdin);
scanf("%d%d",&n,&m);
for (i=1;i<=n;i++)
{
fa[i]=i;
rx[i]=ry[i]=0;
}
for (i=1;i<=m;i++)
{
scanf("%d%d%d %c",&x[i],&y[i],&d,&c);
switch(c)
{
case 'W':dx[i]=-d;dy[i]=0;break;
case 'S':dx[i]=0;dy[i]=-d;break;
case 'E':dx[i]=d;dy[i]=0;break;
case 'N':dx[i]=0;dy[i]=d;break;
}
}
scanf("%d",&k);
for (i=1;i<=k;i++)
{
scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].idx);
a[i].n=i;
}
sort(a+1,a+k+1,cmp);
j=1;
for (i=1;i<=k;i++)
{
for (;j<=a[i].idx;j++)
{
fx=find(x[j]);fy=find(y[j]);
fa[fy]=fx;
rx[fy]=rx[x[j]]-rx[y[j]]-dx[j];
ry[fy]=ry[x[j]]-ry[y[j]]-dy[j];
}
if (find(a[i].x)!=find(a[i].y))
ans[a[i].n]=-1;
else
ans[a[i].n]=abs(rx[a[i].x]-rx[a[i].y])+abs(ry[a[i].x]-ry[a[i].y]);
}
for (i=1;i<=k;i++)
printf("%d\n",ans[i]);
}
作者:ascii991
补充:软件开发 , C++ ,