洛谷/CF Rudolf and 121

鲁道夫有一个由 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;
}