poj 1984 并查集(在线查询曼哈顿距离）

来源：互联网发布：传智播客php教程编辑：程序博客网时间：2024/05/02 02:46

题意：一个图中，只有上下左右四个方向的边。给出这样的一些边，求任意指定的2个节点之间的曼哈顿距离。查询时还有第三个参数k，表示只知道图中前k条边的信息时查询，此时如果不能确定，则输出-1。

思路：并查集。时时维护相对于并查集树顶结点的坐标偏移cx和cy两个数组。合并的时候仅仅更新树顶结点，其他结点在find操作时更新。

#include <cstdio>#include <cstring>#include <cstdlib>#include <cmath>#include <algorithm>#include <queue>using namespace std;#define INF 0x3fffffff#define clr(s,t) memset(s,t,sizeof(s))#define N 40005int f[N];struct edge{    int a,b,dx,dy;}e[N];struct question{    int a,b,id,init;}q[10005];int n,m,qnum,res[10005],cx[N],cy[N];int cmp(question x,question y){    return x.id < y.id;}int find(int x){    if(f[x] == x)        return x;    int tmp = f[x];    f[x] = find(f[x]);    cx[x] += cx[tmp];//每一次find操作，结点x都更新到其最新的坐标上    cy[x] += cy[tmp];    return f[x];}int main(){    int i,j,c,xx,yy;    char ori;    scanf("%d %d",&n,&m);    clr(cx, 0);    clr(cy, 0);    for(i = 1;i<=n;i++)        f[i] = i;    for(i = 1;i<=m;i++){        scanf("%d %d %d %c",&e[i].a,&e[i].b,&c,&ori);        e[i].dx = e[i].dy = 0;        if(ori == 'E')            e[i].dx = c;        else if(ori == 'W')            e[i].dx = -c;        else if(ori == 'N')            e[i].dy = c;        else            e[i].dy = -c;    }    scanf("%d",&qnum);    for(i = 1;i<=qnum;i++){        scanf("%d %d %d",&q[i].a,&q[i].b,&q[i].id);        q[i].init = i;    }    sort(q+1,q+qnum+1,cmp);    for(i = j = 1;i<=qnum;i++){        for(;j<=q[i].id;j++){            xx = find(e[j].a);            yy = find(e[j].b);            if(xx == yy)//如果已经在一个区域了，那么他们的相对坐标是确定的，不用再次进行处理                continue;            f[yy] = xx;            cx[yy] = cx[e[j].a]-cx[e[j].b]+e[j].dx;//只对yy所在集合的顶点结点进行处理，剩下的结点在进行find时处理            cy[yy] = cy[e[j].a]-cy[e[j].b]+e[j].dy;        }        if(find(q[i].a) != find(q[i].b))            res[q[i].init] = -1;        else            res[q[i].init] = abs(cx[q[i].a]-cx[q[i].b])+abs(cy[q[i].a]-cy[q[i].b]);    }    for(i = 1;i<=qnum;i++)        printf("%d\n",res[i]);    return 0;}

0 0