HDU-4879-ZCC loves march(map+set+并查集)

标签：des   style   blog   http   color   os   io   2014   

 5 3
1 3
2 1
2 2
2 3
3 2
6
Q 1
L 0 2
L 5 2
Q 5
R 3 1
Q 3 

 1
1
7 

The input after decode: Q 1 L 1 2 L 4 2 Q 4 R 2 1 Q 2 

思路：分别用map来维护与x坐标平行和垂直的线上面的集合，集合里面直接存对应元素的下标，并用并查集维护，根节点可以用来表示所在的集合。当移动一个元素时，先找到该元素下标对应的根，根对应的num-1，并把移动的元素插到新的位置上，他的根就是他自己。执行Q操作时，通过map找到同一行和同一列的所有集合进行计算，再把这些计算过的集合删掉，同时下标指向新的根（即移动之后形成的新的集合）。

#include <cstdio>
#include <cmath>
#include <set>
#include <map>
#define LL long long
#define mod 1000000007
using namespace std;

struct TROOP{
LL x,y,num;
TROOP(){}
TROOP(LL nx,LL ny,LL nnum){x=nx,y=ny,num=nnum;}
}troop[200005];

map<LL,set<LL> >MX;
map<LL,set<LL> >MY;
set<LL>::iterator it;

LL node[200005];

LL findroot(LL x)
{
    if(node[x]!=x) node[x]=findroot(node[x]);

    return node[x];
}

int main()
{
    LL m,x,y,n,i,t,cnt,root,a,b,ans;
    char s[5];

    while(~scanf("%I64d%I64d",&n,&m))
    {
        MX.clear();
        MY.clear();

        for(i=1;i<=n;i++)
        {
            scanf("%I64d%I64d",&x,&y);

            troop[i].x=x;
            troop[i].y=y;
            troop[i].num=1;

            node[i]=i;

            MX[x].insert(i);
            MY[y].insert(i);
        }

        cnt=n+1;

        ans=0;

        scanf("%I64d",&t);

        while(t--)
        {
            scanf("%s",s);

            if(s[0]=='Q')
            {
                scanf("%I64d",&a);

                a^=ans;

                root=findroot(a);//找到a所在的集合

                x=troop[root].x;
                y=troop[root].y;

                LL num=0;

                ans=0;

                for(it=MX[x].begin();it!=MX[x].end();it++)
                {
                    num+=troop[*it].num;

                    LL temp=abs(troop[*it].y-y);

                    temp%=mod;

                    ans=(temp*temp%mod*troop[*it].num%mod+ans)%mod;

                    node[*it]=cnt;//指向cnt，cnt是执行Q操作之后新的根，用来标记新的集合

                    MY[troop[*it].y].erase(*it);//*it已经计算过，从MY[]集合里删掉，避免重复计算
                }

                for(it=MY[y].begin();it!=MY[y].end();it++)
                {
                    num+=troop[*it].num;

                    LL temp=abs(troop[*it].x-x);

                    temp%=mod;

                    ans=(temp*temp%mod*troop[*it].num%mod+ans)%mod;

                    node[*it]=cnt;//同理

                    MX[troop[*it].x].erase(*it);//同理
                }

                node[cnt]=cnt;//指向自己，别忘了
                troop[cnt]=TROOP(x,y,num);//执行Q操作之后形成的新集合
                MX[x].clear();
                MY[y].clear();
                MX[x].insert(cnt);//在目标集合中插入
                MY[y].insert(cnt);

                cnt++;

                printf("%I64d\n",ans);
            }
            else
            {
                scanf("%I64d%I64d",&a,&b);

                a^=ans;

                root=findroot(a);//找到a所在的集合，即a的根节点

                x=troop[root].x;
                y=troop[root].y;

                troop[root].num--;//集合里的计数减一

                if(!troop[root].num)//如果集合的计数为0则把该集合删掉
                {
                    MX[x].erase(root);
                    MY[y].erase(root);
                }

                if(s[0]=='U')
                {
                    troop[a]=TROOP(x-b,y,1);

                    node[a]=a;//a指向自己，作为新的根

                    MX[x-b].insert(a);//在目标位置插入
                    MY[y].insert(a);
                }
                else if(s[0]=='L')//以下同理
                {
                    troop[a]=TROOP(x,y-b,1);

                    node[a]=a;

                    MX[x].insert(a);
                    MY[y-b].insert(a);
                }
                else if(s[0]=='D')
                {
                    troop[a]=TROOP(x+b,y,1);

                    node[a]=a;

                    MX[x+b].insert(a);
                    MY[y].insert(a);
                }
                else if(s[0]=='R')
                {
                    troop[a]=TROOP(x,y+b,1);

                    node[a]=a;

                    MX[x].insert(a);
                    MY[y+b].insert(a);
                }
            }
        }
    }
}

HDU-4879-ZCC loves march(map+set+并查集),布布扣,bubuko.com

HDU-4879-ZCC loves march(map+set+并查集)

标签：des   style   blog   http   color   os   io   2014   

原文地址：http://blog.csdn.net/faithdmc/article/details/38223667