1. 程式人生 > >Gym - 101775J Straight Master —— 差分陣列

Gym - 101775J Straight Master —— 差分陣列

題意:

問能不能通過操作多次將長度為3~5的區間都加1得到給定的數列

思路:

利用差分陣列的思想,將區間X,Y都加1可以變成在X處加1,在Y+1處減1,這樣我們可以通過拿這一項減前一項得到給定數列的差分陣列,因為每個區間長度至少要是3,所以在每個大於0的位置,找它加3位置後面是否有足夠的負數與其對應,一旦出現有數不能變成0,就肯定不能形成

#include <iostream>
#include <cstdio>
#include <cmath>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <string>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define max_ 100010
#define mod 1000000007
#define inf 0x3f3f3f3f
int casnum=1;
int n;
ll num[200010];
ll ans[200010];
int main(int argc, char const *argv[]) {
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(num,0,sizeof num);
        memset(ans,0,sizeof ans);
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&num[i]);
        }
        ans[n+1]=-num[n];
        for(int i=n;i>=1;i--)
        {
            ans[i]=num[i]-num[i-1];
        }
        ll sum=ans[1]+ans[2]+ans[3];
        int f=1;
        int j=1;
        for(int i=1;i<=n+1;i++)
        {
            if(ans[i]>0)
            {
                j=max(j,i+3);
                while(j<=n+1)
                {
                    if(ans[j]<0)
                    {
                        if(ans[i]>=abs(ans[j]))
                        {
                            ans[i]+=ans[j];
                            ans[j]=0;
                        }
                        else
                        {
                            ans[j]+=ans[i];
                            ans[i]=0;
                        }
                    }
                    if(ans[i]==0)
                    break;
                    else
                    j++;
                }
            }
            if(ans[i]!=0)
            {
                f=0;
                break;
            }
        }
        if(f)
        printf("Case #%d: Yes\n",casnum++ );
        else
        printf("Case #%d: No\n",casnum++ );
    }
    return 0;
}