算法初步----哈希习题(算法笔记-分组统计)

本题哈希的思想就是:以输入的数字为下标,来获取该数的信息,利用较大的数组完成。简单算法题多为这种思想,可以通过算出最大数来达到减小数组的目的。

//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

猜你喜欢

转载自blog.csdn.net/Capricorn_devil/article/details/86816729