《算法笔记》4.2小节——算法初步-＞哈希 问题 B: 分组统计

题目描述

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

输入

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

输出

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

样例输入 Copy

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

样例输出 Copy

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

代码：

思想中尽管用到了散列，但我觉得应该还有更好的方法。后面用到二重循环，感觉时间复杂度不算低，但暂时没有写出更好的代码了。codeup上通过是没问题的。

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

int main(){
	int m;
	while(scanf("%d",&m)!=EOF){
		int i,j,k;
		for(i=0;i<m;i++){
			int n;
			scanf("%d",&n);
			int data[105];    
			for(j=0;j<n;j++){
				scanf("%d",&data[j]);  //把输入的数据保存在data数组中 
			}
			
			int group[105]={0};
			int group_and_data[105][10000]={0};

			for(j=0;j<n;j++){          //思考如何把数据放到分组中，并表示出现的次数。 
				scanf("%d",&group[j]); //把分组号保存在group数组中   
				group_and_data[group[j]][data[j]]++;//把该组号的该数据做计数+1 
				
			}
			sort(data,data+n);
			sort(group,group+n);
			
			for(k=0;k<n;k++){
				if(k==0||group[k-1]!=group[k]) {
					printf("%d={",group[k]);
					for(j=0;j<n;j++){
						if(j==0||data[j-1]!=data[j]){
							printf("%d=%d",data[j],group_and_data[group[k]][data[j]]);
							if(j!=n-1&&data[j]!=data[n-1]) printf(",");
						}							
					}
					printf("}\n");
								
				}
			}			
		}
	}
	return 0;
}