北理复试上机 分组统计

题目复制于牛客网

题目描述

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

输入描述:

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

输出描述:

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

输入

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

输出

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

解法：

第一次做写了七八十行，死活调不出来，于是转去刷pat-basic和leetcode，一个月之后想起来再重新做了一遍，二十分钟不到就搞定了...

总共代码不到40行... 强 力扣 强

哈希表存储分组和元素的映射关系，关系为

map<int,vector<int>> mp;

每一个分组对应一个由其对应元素组成的数组

 

使用集合存储所有的元素种类

unordered_set<int> datatype;

不需要排序，因此可以使用unordered_set来节省时间

存储好后按序输出即可

代码如下：

#include<bits/stdc++.h>
using namespace std;
void show(vector<int> v,unordered_set<int> datatype){
    map<int,int> mm;
    for(auto it : datatype) 
        mm[it] = 0;
    for(int i = 0;i < v.size();i++) 
        mm[v[i]]++;
    for(auto it = mm.begin();it != mm.end();it++){
        if(it != mm.begin()) 
            cout << ",";
        cout << it->first << "=" << it->second;
    }
}
int main(){
    int n,m;
    cin >> n;
    while(n--){
        cin >> m;
        unordered_set<int> datatype;
        int data[m],group;
        for(int i = 0;i < m;i++) {
            cin >> data[i];
            datatype.insert(data[i]);
        }
        map<int,vector<int>> mp;
        for(int i = 0;i < m;i++){
            cin >> group;
            mp[group].push_back(data[i]);
        }
        for(auto it = mp.begin();it != mp.end();it++){
            cout << it->first << "={";
            show(it->second,datatype);
            cout << "}" << endl;
        }
    }
    return 0;
}