Flip the Bits(思维+标记)

最新推荐文章于 2021-04-08 17:18:26 发布

小菜鸡加油

最新推荐文章于 2021-04-08 17:18:26 发布

阅读量216

点赞数

分类专栏：思维

本文链接：https://blog.csdn.net/zstuyyyyccccbbbb/article/details/115430439

版权

思维专栏收录该内容

514 篇文章 4 订阅

订阅专栏

该博客主要讨论了一种字符串处理策略，通过后缀枚举和前缀标记的方法检查两个字符串是否可以通过仅修改一个字符相等。代码实现中涉及了动态维护前缀0和1计数的技巧，并在O(n)时间内完成判断。这是一个关于字符串算法和数据结构的问题。

摘要由CSDN通过智能技术生成

https://codeforces.com/contest/1504/problem/B

思路：后往前枚举前缀，打标记代替修改即可。

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=3e5+1000;
typedef int LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
char s1[maxn],s2[maxn];
LL sum0[maxn],sum1[maxn];
int main(void){
   cin.tie(0);std::ios::sync_with_stdio(false);
   LL t;cin>>t;
   while(t--){
     LL n;cin>>n;
     for(LL i=0;i<=n+10;i++){
        sum0[i]=sum1[i]=0;
     }

//     for(LL i=1;i<=n;i++){
//        cout<<sum0[i]<<" ";
//     }
//     cout<<"\n";
//     for(LL i=1;i<=n;i++){
//        cout<<sum1[i]<<" ";
//     }
//     cout<<"\n";
     cin>>(s1+1);
     cin>>(s2+1);
     for(LL i=1;i<=n;i++){
        sum0[i]=sum0[i-1]+(s1[i]=='0');
        sum1[i]=sum1[i-1]+(s1[i]=='1');
     }
     LL lazy=1;
     bool flag=1;
     for(LL i=n;i>=1;i--){
        if(lazy==1){
            if(s1[i]==s2[i]) continue;
            else{
              if(sum0[i]!=sum1[i]){
                 flag=0;
                 break;
              }
              lazy=0;
            }
        }
        else if(lazy==0){
            if(s1[i]!=s2[i]) continue;
            else{
                if(sum0[i]!=sum1[i]){
                    flag=0;
                    break;
                }
                lazy=1;
            }
        }
     }
     if(flag){
        cout<<"YES"<<"\n";
     }
     else{
        cout<<"NO"<<"\n";
     }
   }
   return 0;
}