Same Differences——STL容器中的unordered_map学习

本文链接：https://blog.youkuaiyun.com/qq_51455933/article/details/116450763

这篇博客主要介绍了如何利用C++的unordered_map容器处理输入数据，并通过迭代器遍历计算特定数值。文章中展示了如何避免整型溢出和类型转换错误，对于理解和应用C++数据结构解决实际问题有一定帮助。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

    #include <iostream>
    #include <string>
    #include <unordered_map>
    #include <algorithm>
    using namespace std;
    #define ll long long
    //1.开ll，不然会WA
    unordered_map<ll, ll> vis;
    //2.假定这里写int，int,后面需要类型转换，否则会爆int
    int main() {
        int t;
        cin >> t;
        while (t--) {
            vis.clear();
            int n;
            cin >> n;
            ll a;
            for (int i = 1; i <= n; i++) {
                cin >> a;
                vis[a - i]++;
            }
            ll ans = 0;
            unordered_map<ll, ll>::iterator it;
            //3.开迭代器
            for (it = vis.begin(); it != vis.end(); it++) {
            //4.迭代器遍历是使用地址遍历
            //5.指针不能用<，因此写成it!=vis.end()
                ans += it->second * (it->second - 1) / 2;
            }
            cout << ans << endl;
        }
        return 0;
    }

需类型转换的代码

  #include <iostream>
    #include <string>
    #include <unordered_map>
    #include <algorithm>
    using namespace std;
    #define ll long long
    unordered_map<int, int> vis;
    int main() {
        int t;
        cin >> t;
        while (t--) {
            vis.clear();
            int n;
            cin >> n;
            ll a;
            for (int i = 1; i <= n; i++) {
                cin >> a;
                vis[a - i]++;
            }
            ll ans = 0;
            unordered_map<int, int>::iterator it;
            for (it = vis.begin(); it != vis.end(); it++) {
                ans += (ll)it->second * (ll)(it->second - 1) / 2;
                //或者这样写： ans += (ll)it->second * (it->second - 1) / 2;
            }
            cout << ans << endl;
        }
        return 0;
    }