并查集题目

PAT1114-题目：

给出n组：
faID moID K kidsID num total

output：num组家庭
输出每组家庭 ——
smallestID M人 AVG(num) //降序 AVG(total) //avg（num）并列则升序

思路:

cnt[],
cnt_num[],
cnt_total[],

最后遍历cnt[]再对cnt_num[],cnt_total[],操作。

#include<bits/stdc++.h>
#define MAXN 10005
using namespace std;

int pre[MAXN], cnt[MAXN], cnt_num[MAXN], cnt_total[MAXN], num[MAXN], total[MAXN];
bool exist[MAXN];

int fd(int root){
    int son, temp;
    son = root;
    while(root != pre[root])
        root = pre[root];
    while(son != pre[root]){
        temp = pre[son];
        pre[son] = root;
        son = temp;
    }
    return root;
}

void Union(int r1, int r2){
    int root1 = fd(r1);
    int root2 = fd(r2);
    int minn = root1 < root2?root1:root2;   //合并时要求处理
    int man = root1 > root2?root1:root2;
    if(root1 != root2){
        pre[man] = minn;
    }
}

int main(){
    int n, fid, mid, k, id, are, num, maxn = 0;
    for(int i = 0; i < 1005; i++){
        pre[i] = i;
    }

    while(n--){
        cin >> fid >> mid >> k;
        id = (fid < mid)?fid:mid;
        maxn = maxn > fid?maxn:fid;
        maxn = maxn > mid?maxn:mid;
        exist[fid] = exist[mid] = true;
        Union(fid, mid);
        for(int i = 0; i < k; i++){
            cin >> k;
            exist[k] = true;
            maxn = maxn > k?maxn:k;
            Union(k, id);
        }
        cin >> num >> are;
        num[id]=num;
        total[id]=are;
    }
    for(int i = 0; i  <= maxn; i++){
        if(exist[i] == true){
            int root = fd(i);
            cnt[root]++;
            cnt_num[root]+=num[i];
            cnt_num[root]+=total[i];
        }
    }

    //输出要求处理--------怎么操作？
    for(int i = 0; i  <= maxn; i++){
        if(exist[i] == true && cnt[i]){

        }
    }

}

思路改正：

并查集没错
两个结构体数组，一个data接受数据，顺便实现并查集union；
一个ans输出最后的答案，

计算家庭人数：visit标记出现过的节点；
对于每个节点的父节点：people++， flag = true
家庭个数 cnt = true个数
排序后输出前cnt个即为答案