Atcoder166E this Message Will Self-Destruct in 5s

思维题
题意:|i-j| = h_i+h_j
这样看的话不太好做,最开始以为是前缀的题.看了题解之后才发现移项之后非常简单
①如果i>j 则 i-h_i = j+h_j
②如果j>i 则 j-h_j= i +h_i
可以看出,这个式子的本质就是较大的编号减去他本身的高度,等于较小的人的编号加上他自身的高度,所以我们可以在读入的过程中同时计算i+h_i的值,然后每次读入,找到在i之前的所有j+h_j = i-h_i的个数 (1 <=j < i),加入答案中.
要注意的一点是,数组界限的问题.(RE)
起初看题解要计算i+h_i,i-h_i觉得自己懂了,上手操作发现A_i的数据范围是1e9,数组开不了那么大,百度的题解发现基本都只有2e5左右,百思不得其解,后来想通了,因为两个数的差最多是2e5,如果hi太大可以直接忽略.所以数组空间被剧烈压缩,但还是RE了三次,检查的时候发现i-x有可能小于0,这种情况下说明这个人不满足条件,应该直接跳过去.这样就过了!!!.

#include<bits/stdc++.h>

using namespace std;

int a[4100000];
int main(){
	int n;
	long long ans = 0;
	cin >> n;
	for(int i = 1;i <= n;i++)
	{   int x;
		scanf("%d",&x);
		if(i>=x)
		ans+=a[i-x];
		if(i+x<2*n)
		a[i+x]++;
	}
	cout<<ans<<endl;
	return 0; 
}