题解 | #下标数组#

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

int main() {
    int n,m;
    cin>>n;
    unordered_map<int,int> mp;
    long long ans=0;
    for(int i=1;i<=n;i++){
        cin>>m;
        if(mp.count(i-m)) ans+=mp[i-m];
        mp[m-i]++;
    }
    cout<<ans<<endl;
    return 0;
}
// 64 位输出请用 printf("%lld")

a_i+a_j=i+j可以转化为a_i-i=-(a_j-j)。因此在哈希表中仅需存储a_i-i以及它出现的次数，对于a_i-i需要找到i-a_i出现的个数累加到输出即可。