[swustoj 1091] 土豪我们做朋友吧

土豪我们做朋友吧(1091)

问题描述：

人都有缺钱的时候，缺钱的时候要是有个朋友肯帮助你，那将是一件非常幸福的事情。有N个人（编号为1到N），一开始他们互相都不认识，后来发生了M件事情，事情分为2个种类，1:A和B成为了朋友，并且A的所有朋友都成了B的朋友，B的所有朋友也都成了A的朋友。2：A缺钱了，请求帮助，他需要向他朋友中钱最多的请求帮助，若不止一位，选择编号最小的。

输入：

多组测试数据，每组第一行是整数N（2<=N<=100000），表示有N个人，第二行N个数据，依次表示每个人有多少钱，第三行是M（1<=M<=100000），表示发生了M个事件。接下来是M行，首先输入事件种类P（1或者2），然后是对应的两个或者一个整数A,B或者A。数据保证都在32位以内。

输出：

对于每一个事件2，输出A需要请求的人的编号，若没有人能够帮助他，输出"NO ONE CAN HELP!"。

样例输入

3
1 2 3
3
2 1
1 1 3
2 1

样例输出

NO ONE CAN HELP!

3

 

并查集、维护最大和次大就是了

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
#define INF 0x3f3f3f3f
#define N 100010

int n,m;
int a[N];
int f[N];
pair<int,int> mx[N]; //最大的值、人
pair<int,int> cx[N]; //次大的值、人
pair<int,int> tmp[5];

bool cmp(pair<int,int> a,pair<int,int> b)
{
    if(a.first!=b.first) return a.first>b.first;
    return a.second<b.second;
}
void init()
{
    for(int i=1;i<=n;i++)
    {
        f[i]=i;
        scanf("%d",&mx[i].first);
        mx[i].second=i;
        cx[i].first=-1;
        cx[i].second=-1;
    }
}
int Find(int x)
{
    if(x!=f[x]) f[x]=Find(f[x]);
    return f[x];
}
void UN(int x,int y)
{
    x=Find(x);
    y=Find(y);
    if(x==y) return;
    f[x]=y;
    tmp[0]=mx[x];tmp[1]=cx[x];
    tmp[2]=mx[y];tmp[3]=cx[y];
    sort(tmp,tmp+4,cmp);
    mx[x]=mx[y]=tmp[0];
    cx[x]=cx[y]=tmp[1];
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        init();
        scanf("%d",&m);
        while(m--)
        {
            int op,a,b;
            scanf("%d%d",&op,&a);
            if(op==1)
            {
                scanf("%d",&b);
                UN(a,b);
            }
            else
            {
                b=Find(a);
                if(mx[b].first!=-1 && mx[b].second!=a)
                    printf("%d\r\n",mx[b].second);
                else if(cx[b].first!=-1)
                    printf("%d\r\n",cx[b].second);
                else
                    printf("NO ONE CAN HELP!\r\n");
            }
        }
    }
    return 0;
}