kls与flag（map）

题目传送门

这道题还挺搞笑的，\(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;
}