Straight Master Gym-101775J (差分)

题目来源 Straight Master

题意

有n种扑克牌，每种扑克牌有ai张，每次可以打出3到5张连续的牌作为顺子，问这副牌能不能用顺子全打出来

思路

换一个思路，给定一个长度为0的序列，每次可以选择长度为3,4,5的区间并将这个区间内的数全部加一，最终可以得到一个新的序列，问这个序列的每个数分别是多少，这个序列就是给定的n种扑克牌。

对于这个问题，可以用差分的思想，对于区间[L, R]，可以开一个新的数组b，这个区间加一后可以认为是b[L]+=1, b[R+1]-=1, b的前缀和即为对应的数字。

原来那个问题就可以转化为给你一个序列，问这个序列可不可以由上面的操作得到。也可以构建一个差分数组b，其中b[i] = a[i]-a[i-1]。如果这个b数组对于每个相邻距离大于等于3的b[i] 和 b[j] (j>=i+3)，如果每一对的和加起来等于0，则给定的数列是可以得到的，否则就无法得到。

代码

#include <bits/stdc++.h>
using namespace std;
int n, a[200005], b[200005];
int main()
{
    int t;
    scanf("%d", &t);
    for (int cas = 1; cas <= t; ++cas)
    {
        memset(a, 0, sizeof(a));
        memset(b, 0, sizeof(b));

        scanf("%d", &n);
        for (int i = 0; i < n; ++i) scanf("%d", a + i);

        b[0] = a[0];
        for (int i = 1; i < n; ++i) b[i] = a[i] - a[i - 1];
        b[n] = -a[n - 1];

        bool ok = true;
        if (b[0] < 0 || b[1] < 0 || b[2] < 0)   ok = false;
        else
        {
            long long sum = 0;
            for (int i = 0; i <= n; ++i)
            {
                if (b[i] > 0)   sum += b[i];
                int p = i + 3;
                if (p > n)  break;
                if (b[p] < 0)
                {
                    sum += b[p];
                    b[p] = 0;
                }
                if (sum < 0)    break;
            }
            if (sum != 0)   ok = false;
        }
        printf("Case #%d: %s\n", cas, ok ? "Yes" : "No");
    }
    return 0;
}

这个题真的是。。。输出格式弄错了wa了好几发，然后判断前缀和满不满足条件的时候写到了花括号里。。。智障。。。