本题哈希的思想就是:以输入的数字为下标,来获取该数的信息,利用较大的数组完成。简单算法题多为这种思想,可以通过算出最大数来达到减小数组的目的。
//http://codeup.cn/problem.php?cid=100000582&pid=1
#include <iostream>
#include <algorithm> //max_element,unique_copy
#include <cstring>
#include <vector>
using namespace std;
int main() {
int M;
while(cin>>M) {
while (M--) {
int N; //输入数据的个数
cin >> N;
int nums[N], cls[N]; //数据和类别
for (int i = 0; i < N; i++)
cin >> nums[i];
for (int i = 0; i < N; i++)
cin >> cls[i];
// 数据的最大值和类别
int max_cls = *max_element(cls, cls + N);
int max_num = *max_element(nums, nums + N);
// hash二维数组计数
int hashTable[max_cls + 1][max_num + 1];
// 全0初始化
memset(hashTable, 0, sizeof(hashTable));
for (int i = 0; i < N; i++)
hashTable[cls[i]][nums[i]]++;
//类排序去重
vector<int> v2;
sort(cls, cls+N);
unique_copy(cls, cls + N, back_inserter(v2));
//nums排序去重
vector<int> v;
sort(nums, nums + N);
unique_copy(nums, nums + N, back_inserter(v));
for (int i = 0; i < v2.size(); i++) {
cout << v2[i] << "={";
for (int j = 0; j < v.size(); j++) {
cout << v[j] << '=' << hashTable[v2[i]][v[j]];
if (j < v.size() - 1) cout << ',';
}
cout << '}' << endl;
}
}
}
return 0;
}
代码地址:https://github.com/CapricornHearth/algorithm/tree/master/%E5%93%88%E5%B8%8C