分组统计【北京理工大学】

题目链接

注意

对于分组为空的小组要直接跳过

#include<bits/stdc++.h>
using namespace std;
int arr[105];
int main(){
	int m,n,x; 
	while(cin>>m){
		while(m--){
			cin>>n;
			vector<int> G[105];
			for(int i=0;i<105;i++){
				G[i].clear();
			}
			set<int> st;
			st.clear();
			for(int i=0;i<n;i++){
				cin>>arr[i];
				st.insert(arr[i]);
			}
			int k=0;
			for(int i=0;i<n;i++){
				cin>>x;
				G[x].push_back(arr[i]);
				if(x>k) k=x;
			}
			int hsh[10001];
			for(int i=1;i<=k;i++){
				memset(hsh,0,sizeof(hsh));
				if(G[i].size()==0) continue; //很关键
				for(int j=0;j<G[i].size();j++){
					int v=G[i][j];
					hsh[v]++;
				}
				printf("%d={",i);
				int t=0;
				for(auto it:st){
					printf("%d=%d",it,hsh[it]);
					if(t!=st.size()-1) printf(",");
					else printf("}\n");
					t++;
				}
			}
		}
	}
	
	return 0;
}