Flip the Bits(思维+标记)

该博客主要讨论了一种字符串处理策略,通过后缀枚举和前缀标记的方法检查两个字符串是否可以通过仅修改一个字符相等。代码实现中涉及了动态维护前缀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;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值