Balanced Bitstring


题意:判断长度为n的串中的每个长度为k的子串是否0和1相等,字符串中的问号既可以是1也可以是0。
解题思路:
例如n=10 ,k=4 对于 s[0~3] 和 s[1~4] 我们可以发现1~3的部分是相同的,那么不难发现0和4相同,同理可得s[2]和s[5]也是相同的因此若该串合理就一定是k循环节子串,最后我们只需要判断前k个字母是否可以构造出一半1和一半0 ,如果1或者0的个数已经大于一半了一定构造不出来,否则一定可以构造出来。

代码:

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=300010;
bool f[N];
// x _ _ _ _    _ _ _ _ x
string s;
int main()
{
   
    // ?00 000 100
    int _;
    cin>>_;
    while(_--)
    {
   
      int n,k;
      cin >> n >> k;
      cin>>s;
      bool flag=true;
      for(int i=0;i<n;i++)
      {
   
          if(s[i]=='?' && s[i%k]=='?')
            continue;
         if(s[i]=='?' && s[i%k] != '?')
            continue;
         if(s[i]!='?' && s[i%k] == '?')
            {
   
                s[i%k] =s[i];
                continue;
            }
            if(s[i]!='?' && s[i%k]!='?' && s[i]!=s[i%k])
            {
   
                flag=false;
                break;
            }
      }
      int s1=0;
      int s2=0;
      if(!flag)
      {
   
        cout<<"NO"<<endl;
        continue;
      }
      for(int i=0;i<k;i++)
      {
   
        s1+=(s[i]=='0');
        s2+=(s[i]=='1');
      }
      if(max(s1,s2)>k/2)    cout<<"NO"<<endl;
      else      cout<<"YES"<<endl;
    }
}
全部评论

相关推荐

totoroyyw:千年老妖😂
投递华为等公司10个岗位
点赞 评论 收藏
分享
把球:这个听过,你加了就会发现是字节的hr
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务