题解 | 谐距下标对

#include <iostream>
#include <unordered_map>
using namespace std;

int main() {
    int n;
    cin >> n;
    unordered_map<int, int> mp;
    for(int i=1; i<=n; ++i){
        int ai;
        cin >> ai;
        ++mp[ai-i];
    }
    long long ans=0;
    for(auto it:mp){
        ans += (long long )it.second*(it.second-1)/2 ;
    }
    cout << ans << endl;
}
// 64 位输出请用 printf("%lld")