PAT练习 分组统计

PAT练习 分组统计

Description

先输入一组数，然后输入其分组，按照分组统计出现次数并输出，参见样例。

Input

输入第一行表示样例数m，对于每个样例，第一行为数的个数n，接下来两行分别有n个数，第一行有n个数，第二行的n个数分别对应上一行每个数的分组，n不超过100。

Output

输出m行，格式参见样例，按从小到大排。

Sample Input

1
7
3 2 3 8 8 2 3
1 2 3 2 1 3 1

Sample Output

1={2=0,3=2,8=1}
2={2=1,3=0,8=1}
3={2=1,3=1,8=0}

C++实现

思路：为了对数字和组别去重，分别声明两个set用以保存有哪些数字及哪些组。题目要求统计每一组的每一个数字出现的次数，因此需要一个双重map

#include <stdio.h>
#include <algorithm>
#include <map>
#include <set>
using namespace std;

void show(map<int,int> m){
    map<int,int>::iterator it;
    it=m.begin();
    printf("%d=%d",it->first,it->second);
    for(it++;it!=m.end();it++){
        printf(",%d=%d",it->first,it->second);
    }
}

int main(){
    int m,n,nums[110],group[110];
    map<int,map<int,int> > mp;
    set<int> set_nums;
    set<int> set_group;
    scanf("%d",&m);
    while(m--){
        scanf("%d",&n);
        for(int i=0;i<n;i++){
            scanf("%d",&nums[i]);
            set_nums.insert(nums[i]);
        }
        for(int i=0;i<n;i++){
            scanf("%d",&group[i]);
            set_group.insert(group[i]);
        }
        for(set<int>::iterator it_group=set_group.begin();it_group!=set_group.end();it_group++){
            for(set<int>::iterator it_nums=set_nums.begin();it_nums!=set_nums.end();it_nums++){
                mp[*it_group][*it_nums]=0;
            }
        }
        for(int i=0;i<n;i++){
            mp[group[i]][nums[i]]++;
        }
        for(map<int,map<int,int> >::iterator it=mp.begin();it!=mp.end();it++){
            printf("%d={",it->first);
            show(it->second);
            printf("}\n");
        }
        mp.clear();
        set_nums.clear();
        set_group.clear();
    }
}