[Swust OJ 1091]--土豪我们做朋友吧(并查集,最值维护)

 

题目链接：http://acm.swust.edu.cn/problem/1091/

 

Time limit(ms): 1000　　　　　　Memory limit(kb): 32768

 

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

Description

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

Input

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

Output

1 2 3 4 5 6 7

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

Sample Input

1 2 3

NO ONE CAN HELP! 3

Sample Output

输出换行请使用\r\n

 

解题思路：直接的一个并查集能过所有的数据，但是数据量过大，单纯的并查集会直接超时；

　　　　　那么在使用并查集将最大与次大元素找出，然后维护就可以了

（之所以找最大和次大，在并查集合并过程中初始化最大元素为自己，恩，你懂得~~）

代码如下：

#include<iostream>
#include<cstdlib>
#include<cstring>
using namespace std;
int n, m, a, b, c, f[100001], val[100001];
struct tuhao{
    int pos, money;
};
struct node{
    tuhao max[4];
}x[100001];
int cmp(const void *x, const void *y){
    if ((*(tuhao *)x).money == (*(tuhao *)y).money)
        return (*(tuhao *)x).pos - (*(tuhao *)y).pos;
    return (*(tuhao *)y).money - (*(tuhao *)x).money;
}
void init(){
    memset(x, 0, sizeof(x));
    for (int i = 1; i <= n; i++){
        x[i].max[0].money = val[i];
        x[i].max[0].pos = f[i] = i;
 
    }
}
int findset(int k){
    int i = k, j = k, r;
    while (i != f[i])
        i = f[i];
    while (j != i){
        r = f[j];
        f[j] = i;
        j = r;
    }
    return i;
}
void mergy(int a, int b){
    int tx = findset(a), ty = findset(b), i;
    if (tx != ty){
        f[ty] = tx;
        for (i = 2; i < 4; i++){
            x[tx].max[i].pos = x[ty].max[i - 2].pos;
            x[tx].max[i].money = x[ty].max[i - 2].money;
        }
        qsort(x[tx].max, 4, sizeof(x[tx].max[0]), cmp);
        for (i = 2; i < 4; i++)
            x[tx].max[i].pos = x[tx].max[i].money = 0;
    }
}
void Search(int k){
    int v = findset(k);
    if (!x[v].max[1].pos)
        cout << "NO ONE CAN HELP!\r\n";
    else
        cout << (x[v].max[0].pos == k ? x[v].max[1].pos : x[v].max[0].pos) << "\r\n";
}
int main(){
    while (cin >> n){
        for (int i = 1; i <= n; i++)
            cin >> val[i];
        init();
        cin >> m;
        for (int i = 0; i < m; i++){
            cin >> a;
            switch (a){
            case 1:cin >> b >> c; mergy(b, c); break;
            default:cin >> b; Search(b); break;
            }
        }
    }
    return 0;
}