AtCoder Beginner Contest 166 E.This Message Will Self-Destruct in 5s

AtCoder Beginner Contest 166 E.This Message Will Self-Destruct in 5s

題目鏈接

思維題，題意是找一對 $(i,j)$ 使得 $a_i+a_j=j-i$，我們不妨設 $i<j$，移項得：$a_j-j=-a_i-i$，很容易想到用 $map$ 存數，每次更新答案即可，AC代碼如下：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

main()
{
    int n;
    cin>>n;
    int a[n+1];
    map<int,int>m;
    ll ans=0;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        ans+=m[a[i]-i];
        m[-a[i]-i]++;
    }
    cout<<ans;
    return 0;
}