PAT 1114 Family Property

个人学习记录，代码难免不尽人意

Sample Input:

6666 5551 5552 1 7777 1 100

1234 5678 9012 1 0002 2 300

8888 -1 -1 0 1 1000

2468 0001 0004 1 2222 1 500

7777 6666 -1 0 2 300

3721 -1 -1 1 2333 2 150

9012 -1 -1 3 1236 1235 1234 1 100

1235 5678 9012 0 1 50

2222 1236 2468 2 6661 6662 1 300

2333 -1 3721 3 6661 6662 6663 1 100

Sample Output:

8888 1 1.000 1000.000

0001 15 0.600 100.000

5551 4 0.750 100.000

cpp 复制代码

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
struct node{
    int m=0,a=0;
}Node[10010];
struct result{
    int id;
    int num;
    double avgm,avga;
};
int dad[10010];
int findfather(int n){
    if(dad[n]==n) return n;
    else  findfather(dad[n]);
}
bool cmp(result a,result b){
    if(a.avga!=b.avga) return a.avga>b.avga;
    else return a.id<b.id; //不要忘记加return
}
vector<int> root;
int main(){
    int n;
    scanf("%d",&n);
    for(int i=0;i<10010;i++){
        dad[i]=i;
    }
    for(int i=0;i<n;i++){
        int id,father,mother,k;
        scanf("%d %d %d %d",&id,&father,&mother,&k);
        int arr[k+3];
        arr[0]=id;arr[1]=father;arr[2]=mother;
        for(int j=3;j<k+3;j++){
            scanf("%d",&arr[j]);
        }
        bool flag=true;
        int f;
        for(int j=0;j<k+3;j++){
            if(findfather(arr[j])!=arr[j]&&arr[j]!=-1){
                flag=false;
                f=findfather(arr[j]);
                break;
            }
        }
        if(flag){
            root.push_back(arr[0]);
            for(int j=0;j<k+3;j++){
                if(arr[j]!=-1)
                dad[findfather(arr[j])]=dad[arr[0]];
            }
        }
        else{
            for(int j=0;j<k+3;j++){
                if(arr[j]!=-1)
                dad[findfather(arr[j])]=f;
            }
        }
        int m,a;
        
        scanf("%d %d",&m,&a);
        node no;no.m=m;no.a=a;
        Node[id]=no;
    }
    vector<int> temp;
    for(int i=0;i<root.size();i++){
    	if(findfather(root[i])==root[i])
    	temp.push_back(root[i]);
	}
	root=temp;
//	for(int i=0;i<root.size();i++){
//    	cout << root[i] <<endl;
//	}
//	cout << "end";
    vector<result> v;
    for(int i=0;i<root.size();i++){
        int r=root[i];
        int minid=10010;
        int sm=0,sa=0;
        int num=0;
        for(int j=0;j<10010;j++){
               if(findfather(dad[j])==r){
                   num++;
                   if(j<minid) minid=j;
                   sm+=Node[j].m;
                   sa+=Node[j].a;
               }
        }

        double avgm=sm/(1.0*num),avga=sa/(1.0*num);//这个地方必须将除数变为double再相除，否则得不到正确答案。 
     
        result res;res.id=minid;res.num=num;
        res.avgm=avgm;res.avga=avga;
        v.push_back(res);
    }
    sort(v.begin(),v.end(),cmp);
    printf("%d\n",v.size());
    for(int i=0;i<v.size();i++){
        printf("%04d %d %.3lf %.3lf\n",v[i].id,v[i].num,v[i].avgm,v[i].avga);
    }
}

这道题我采用并查集来做而不是二叉树，因为我考虑到数据量可能比较庞大用二叉树模拟可能会超时。这道题本身不算难，但是暴露出的我的问题还是蛮多的：

①：忘记写return了

②：如果要求double=int/int，最好是将int都转换成double来做。

③：我的做法是无视父母节点和孩子节点，都将他们连接到一起，但是这样做会造成取得了多余的root节点，因此我们在遍历完毕之后需要再循环判断一次这些节点还是不是root节点（不是root节点的点在遍历后father就发生改变了），这样才能保证最后的正确。
这道题还是比较有价值的，值得动手一试。