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
",ans);
	return 0;
}