鲁道夫有一个由 n 个整数组成的数组 a ,元素的编号从 1 到 n 。
在一次操作中,他可以选择一个索引 i ( 2≤i≤n−1 )。 并进行赋值:
鲁道夫可以执行任意次数的操作。任何索引 i都可以使用 0 次或更多次。
他能用此操作使数组中的所有元素都等于零吗?
Input
输入的第一行包含一个整数 t ( 1≤t≤1e4 )。--测试中的测试用例数。
每个案例的第一行包含一个整数 n ( 3≤n≤2⋅1e5 ) - 数组中的元素个数。
每个案例的第二行包含 n 个整数 a1,a2,…,an ( 0≤aj≤1e9 )。 - 数组的元素。
保证所有测试用例中 n 的值之和不超过 2⋅1e5 。
Output
对于每个测试用例,如果可以通过所述操作使数组的所有元素为零,则输出 "是"。否则,输出 "否"。
每个字母可以以任何大小写(小写或大写)输出。例如,字符串 "yEs"、"yes"、"Yes "和 "YES "将被视为肯定答案。
输入数据 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
Note
在第一个例子中,原始数组为 [1,3,5,5,2] ,要使所有元素为零,鲁道夫可以按如下方法操作:
- 在 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] 。
题目讲解
我觉得这应该是个贪心把……
我们可以看出,根据n和t知道这肯定是一个O(nt)的时间复杂度,在t组测试点中,最好使用O(n)。
于是我们可以考虑贪心思路:
首先我们可以发现:
- 每次不能去改最左面和最右面
- 最左边或最右边只能被
和
修改
所以我们可以得出结论:把最左边(最右边)删除时(变成0时)
- a[i+1]-=a[i]*2 (a[i-1]-=a[i]*2)
- a[i+2]-=a[i] (a[i-2]-=a[i])
这样我们可以写成代码:
#include<bits/stdc++.h>
#define int long long
using namespace std;
int a[200005];
signed main()
{
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
int t;
cin>>t;
while(t--)
{
int n,flag=0;
cin>>n;
a[n+1]=0,a[n+2]=0;
for(int i=1;i<=n;++i)
{
cin>>a[i];
}
for(int i=1;i<=n;++i)
{
if(a[i]<0)
{
flag=1;
cout<<"NO\n";
break;
}
a[i+1]-=a[i]*2;
a[i+2]-=a[i];
a[i]-=a[i];
}
if(flag==0 && a[n+1]==0 && a[n+2]==0)
{
cout<<"YES\n";
}
else if(flag==0)
{
cout<<"NO\n";
}
}
//注意好边界情况!!!
return 0;
}