当前位置:编程学习 > C/C++ >>

并查集练习---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++ ,
CopyRight © 2022 站长资源库 编程知识问答 zzzyk.com All Rights Reserved
部分文章来自网络,