郑大第九届校赛-热身赛-zzulioj-1808-小d的翻转游戏

1808: 小d的翻转游戏

Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 143 Solved: 32

SubmitStatusWeb Board
Description

小d同学最近在玩翻转棋的游戏，这个棋类游戏的棋子是黑白两面的，他玩这个游戏已经玩了很久了，他觉得没有什么新意了，所以他打算换一种玩法。首先他把这些黑色和白色的棋子排在了一条直线上，之后就是他考验自己的时候了，他会随机选一个数k,然后每次翻转连续的k个棋子，他的目标是把所有的棋子全部变成白色的。但是他的时间有限，他最多有翻转m次的时间。所以问题来了，他需要你告诉他他能不能在个条件下完成他的最后目标。

Input

输入的第一个整数T，代表样例个数，接下来的会有T组样例。

每组样例的第一行会有一个整数k,m,含义如题意所示（1<= k,m <= 100000）。然后是一个01串，其中0代表白色，1代表黑色。这个01串的长度len不会超过100000。k <= len。

Output

对于每组样例如果能完成目标输出“YES”，不然输出“NO”。

Sample Input

3
1 1
1
2 100
01
3 2
111111
Sample Output

YES
NO
YES

暴力肯定超时，下面的思路来自网络，表示自己当时也没想到优化方法。
对于任意一个数0或1，如果它被翻转了偶数次，那么他还是他本身，否则就相当于翻转了一次。记录下每个元素被翻转的次数。

代码

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<queue>
using namespace std;
const int maxn=100005;
char str[maxn];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int K,M;
        memset(str,'\0',sizeof(str));
        scanf("%d%d%s",&K,&M,str);
        int num=0;//记录翻转次数
        queue<int>q;//下标入队
        bool Is_Right=true;
        for(int i=0,len=strlen(str); i<len; i++)
        {
            while(!q.empty()&&q.front()<i)
                q.pop();
            int flag=str[i]-'0'+q.size();
            if(flag&1)
            {
                num++;
                if(i+K-1>=len)
                {
                    Is_Right=false;
                    break;
                }
                q.push(i+K-1);
            }
        }
        if(num<=M&&Is_Right)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}