题目传送门
这道题还挺搞笑的，\(map\)常数贼大还是把它水过了。
可以发现这道题求的是 \((j>i)j-i=h_i+h_j,j-i=|h_i-h_j|\)的对数。
那么显然，因为高度大于\(0\)，所以一个数对不可能同时满足两条式子，所以可以分开算。
那么进行分类讨论：
\((1)\) \(j-i=h_i+h_j\),等价于\(h_i+i=j-h_j\),那么开一个\(map\),存下所有值的个数，每次都更新答案
\((2)\) \(j-i=h_j-h_i\),等价于\(j-h_j=i-h_i\),同\((1)\)
\((3)\) \(j-i=h_i-h_j\),等价于\(h_i+i=h_j+j\),同\((1)\)
那么如何保证\(j>i\)呢？倒序不就好了吗！
具体详见代码。
\(Code\ Below:\)
#include <bits/stdc++.h>
using namespace std;
const int maxn=200000+10;
int n,m,a[maxn];
long long ans;
map<int,int> mp;
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
mp[n-a[n]]=1;
for(int i=n-1;i>=1;i--){//情况1
ans+=(long long)mp[a[i]+i];
mp[i-a[i]]++;
}
mp.clear();
mp[a[n]-n]=1;
for(int i=n-1;i>=1;i--){//情况2
ans+=(long long)mp[a[i]-i];
mp[a[i]-i]++;
}
mp.clear();
mp[a[n]+n]=1;
for(int i=n-1;i>=1;i--){//情况3
ans+=(long long)mp[a[i]+i];
mp[a[i]+i]++;
}
printf("%lld\n",ans);
return 0;
}