第十三天PAT-A1114 Family Property并查集详细注解

A1114

Descripiton：

家族产权，给出N个人的家族关系即其所拥有的房产数和总面积，求这一系列人属于几个家族，并求其家族人均房产数和人均面积。

思路：

• 由于输入当前人的时候，其孩子的房产数和面积还没输入，所以第一遍输入没法计算房产总数和面积总数，于是我在第一遍输入的时候直接不做集合求并了，而是将所有与当前人为一家人(包括父母节点)的下标存储在结构体vector<int>child中，当所有人的房产数和总面积都输入完毕后，再从头到尾遍历每个人、取出要合并为一家人的下标进行合并；
• id范围是4为整数，其实可以用int存储id，然后数组开1e4个，就不需要手动映射id和下标关系了，题目中有如此明显的提示的时候一定要注意利用，否则会加大解题难度；
• 并查集，值得注意的地方是，题目描述N<=1000，然后maxn就开成1e3+5了，找了好半天…下次遇到段错、运行时错，就先检查最大范围对不对，其次就是注意当下标返回为-1时要进行异常处理；
• 并查集的背诵模板及注解见另一篇博客A1107 Social Clusters

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#include<map>
#include<set>
#include<vector>
#include<queue>
using namespace std;
const int maxn = 1e4+5;
struct person{
    string id;
    int num;
    double avarea,avsets,st,area;
    vector<int>child;	//用于存放一家人的id，包括父母节点
}p[maxn];
map<string, int>toint;	//string型id映射至下标
int fa[maxn], n, eid = 0;	//eid用于分配下标
int getin(string s){		//根据string获得下标
    if(s == "-1") return -1;
    if(toint.count(s) == 0){	//如果该string首次出现
        toint.insert(make_pair(s, eid));	//添加映射
        p[eid].id = s;
        p[eid].num = 1;
        fa[eid] = eid;	//初始化father信息为自己
        return eid++;
    }
    return toint[s];
}
int get(int x){
    if(x == fa[x]) return x;
    return fa[x] = get(fa[x]);
}
void mer(int u, int v){
    if(u == -1 || v == -1) return ;
    u = get(u);
    v = get(v);
    if(u != v){
        if(p[u].id > p[v].id) swap(u,v);	//u为id较小的那个
        fa[v] = u;	
        p[u].num += p[v].num;	//汇总信息到father节点中
        p[u].st += p[v].st;
        p[u].area += p[v].area;
    }
}
bool cmp(int a, int b){
    if(p[a].avarea != p[b].avarea) return p[a].avarea > p[b].avarea;
    return p[a].id < p[b].id;
}
int main()
{
#ifdef ONLINE_JUDGE
#else
    freopen("1.txt", "r", stdin);
#endif // ONLINE_JUDGE
    scanf("%d", &n);
    int cnum;
    string mid, pid1, pid2, cid;
    for(int i = 0; i < n; i++){
        cin>>mid>>pid1>>pid2>>cnum;
        int mnum = getin(mid);
        if(getin(pid1)!=-1)
            p[getin(pid1)].child.push_back(mnum);
        if(getin(pid2)!=-1)
            p[getin(pid2)].child.push_back(mnum);
        for(int j = 0; j < cnum; j++){
            cin>>cid;
            p[mnum].child.push_back(getin(cid));
        }
        scanf("%lf%lf", &p[mnum].st, &p[mnum].area);
    }
    for(int i = 0; i < eid; i++){
        for(int j = 0; j < p[i].child.size(); j++)
            mer(i, p[i].child[j]);
    }
    set<int>mark;	//标记该fahter出现过
    vector<int>ans;
    for(int i = 0; i < eid; i++){
        if(mark.count(get(i)) == 0){
            mark.insert(get(i));
            ans.push_back(get(i));
        }
    }
    for(int i = 0; i < ans.size(); i++){
        int id = ans[i];
        p[id].avarea = (double)p[id].area / p[id].num;
        p[id].avsets = (double)p[id].st / p[id].num;
    }
    sort(ans.begin(), ans.end(), cmp);
    printf("%d\n", ans.size());
    for(int i = 0; i < ans.size(); i++){
        int id = ans[i];
        printf("%s %d %.3f %.3f\n", p[id].id.c_str(), p[id].num, p[id].avsets, p[id].avarea);
    }
    return 0;
}