E--释怀的RT--2023河南萌新联赛第（二）场：河南工业大学

示例1

输入

5
0 1 0 0 10

输出

4

说明

前四个方格被最后一个心岩照亮

示例2

输入

5
0 1 0 0 1

输出

3

说明

第一个方格和第三个方格被第二个格子的心岩照亮，第四个方格被第五个格子的心岩照亮，一共有三个格子被照亮

 解析：

        差分，记录每个位置的变化，不为1则被点亮。

        注意当某个灯的范围很大时，数组越界。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+5;
int n,cnt,f[N],k,t;
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&t);
		if(t==0) continue;
		if(i-1<t){
			f[1]+=1;
			f[i]-=1;
		}
		else{
			f[i-t]+=1;
			f[i]-=1;
		}
		if(n-i<t){
			f[i+1]+=1;
			f[n+1]-=1;
		}
		else{
			f[i+1]+=1;
			f[i+t+1]-=1;	
		}
	}
	for(int i=1;i<=n;i++){
		k+=f[i];
		if(k>0) cnt++;
	}
	cout<<cnt;
	return 0;
}