Rudolf and 121(鲁道夫和 121)

Rudolf and 121(鲁道夫和 121)

原题链接：点击此处跳转

题面翻译

Rudolf 有一个由 $n$ 个整数组成的数组 $a$ ，元素的编号从 $1$ 到 $n$ 。

在每次操作中，他可以任意选择第 $i$ 个元素( $2\le i\le n-1$ ) 并赋值：

• $a_{i-1}=a_{i-1}-1$
• $a_i=a_i-2$
• $a_{i+1}=a_{i+1}-1$

对于任意第 $i$ 个元素，Rudolf 可以不操作，也可以进行任意次操作。

他能用这个运算使数组中的所有元素都等于 $0$ 吗？

输入

第一行

输入一个整数 $t$ ( $1\le t\le 10^4$ ) 作为测试用例数。

对于每个测试用例：

第一行包含一个整数 $n$ ( $3\le n\le 2\cdot 10^5$ ) - 数组的元素个数。

第二行包含 $n$ 个整数 $a_1,a_2,\dots ,a_n$ ( $0\le a_j\le 10^9$ ) 作为数组中的元素。

保证所有测试用例中 $n$ 的和不超过 $2\cdot 10^5$ 。

输出

对于每个测试用例，如果可以通过所述操作使数组的所有元素为 $0$ ，则输出 “YES”，否则，输出 “NO”。

每个字母可以以任何大小写（小写或大写）输出。例如，字符串 “yEs”、“yes”、"Yes "和 "YES "都将被视为肯定答案。

Note

在第一个样例中，原数组为 $[1,3,5,5,2]$ ，要使所有元素为 $0$ ，鲁道夫可以按如下方法操作：

• 在 $i=4$ 处执行操作，数组变为 $[1,3,4,3,1]$ ；
• 在 $i=3$ 处执行操作，数组变为 $[1,2,2,2,1]$ ；
• 在 $i=2$ 处执行操作，数组变为 $[0,0,1,2,1]$ ；
• 在 $i=4$ 处执行操作，数组变为 $[0,0,0,0,0]$ 。

样例输入 #1

7
5
1 3 5 5 2
5
2 4 4 5 1
5
0 1 3 3 1
6
5 6 0 2 3 0
4
1 2 7 2
3
7 1 0
4
1 1 1 1

样例输出 #1

YES
NO
YES
NO
NO
NO
NO

题目大意：

给出T组样例，每组样例其中有n个数，可以进行一下操作

• $a_{i-1}=a_{i-1}-1$
• $a_i=a_i-2$
• $a_{i+1}=a_{i+1}-1$
  对序列进行操作，判断是否能让序列变成全是0的序列

思路 ：

利用贪心的思想，遍历for （int i=1;i<=n-2;i++）将$a_i-2$进行$a_i$次，就是减$2\ast a_i$,然后$a_{i+1}$减$a_i$，途中判断是否有小于0的数字出现，如果有就输出$NO$，最后判断$a_n$和$a_{n-1}$是否为0，如果都是0的话就输出$YES$，否则就是$NO$。
开始没有思路，甚至想用双重循环，然而 $n$ ( $3\le n\le 2\cdot 10^5$ )的范围吓退了我。暴力法果然la ji。

代码:

#include <bits/stdc++.h>
#define int long long 
#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define fi first
#define se second
#define PII pair<int,int>
using namespace std;
const int N = 2e5+5;
int a[N],b[N];
void solve ()
{
	int n;cin>>n;
	for (int i=1;i<=n;i++) 
		cin>>a[i];	
	for (int i=1;i<=n-2;i++)
	{
		if (a[i]<0)
		{
			cout<<"NO"<<'\n';
			return ;
		}
		a[i+1]-=a[i]*2;
		a[i+2]-=a[i];
	}
	if (a[n]==0&&a[n-1]==0)
	cout<<"YES"<<'\n';
	else cout<<"NO"<<'\n';
}

signed main ()
{
	IOS;
	int T =1;
	cin>>T;
	while(T--) solve ();
	return 0;
}