cf-719-D

题目

https://codeforces.com/contest/1520/problem/D

思路

就是找a[i] - i == a[j] - j有多少种组成方式
输入数组时初始化下，哈希表算出相等的有几对。再在值减下标相等的里面组合，公式为Cn2

代码

//a[i] - i == a[j] - j
//Cn2
#include <iostream>
#include <cstring>
#include <algorithm>
#include <unordered_map>
 
using namespace std;
int T, n;
using LL = long long;
const int N = 2e5 + 5;
int a[N];
 
int main()
{
    cin >> T;
    while ( T --)
    {
        //memset(a, 0, sizeof a);
        unordered_map<LL, LL>hash;
        cin >> n;
        for (int i = 0; i < n; i ++ ) {
            scanf("%d", &a[i]);
            a[i] -= i;
            hash[a[i]] ++;
        }
        
        long long ans = 0;
        for(auto [x, y] : hash)
        {
            if(y > 1) ans += y * (y - 1) * 0.5;
        }
        cout << ans << endl;
      
    }

    return 0;
}