B. Flip the Bits

题目链接

思路: 判断a与b是否相等，如果相等输出YES，否则，先求出串a中0（或1）的个数的前缀和，从n-1到0遍历，当反转的次数(反转次数初始为1)是奇数且a[i]!=b[i]，如果sum[i]!=i-sum[i]+1（判断从0~i 1和0的个数是否相等）flag=0，结束循环；否则反转次数加一。当反转的次数是偶数且a[i]==b[i]，在进行类似上面的判断。最后如果flag等于1就输出YES，否则输出NO。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <sstream>
#define ll long long
// #define int long long
#define lowbit(x) ((~x+1)&x)
#define inf 0x3f3f3f3f
using namespace std;
const int N=3e5+10;
int sum[N];


int main() {
    int t;
    cin>>t;
    while(t--) {
        memset(sum,0,sizeof(sum));
        string a,b;
        int n;
        cin>>n;
        cin>>a>>b;
        if(a==b) {
            cout<<"YES"<<endl;
            continue;
        }
        for(int i=0; i<n; i++) {
            if(a[i]=='0') sum[i]=sum[i-1]+1;
            else sum[i]=sum[i-1];
        }
        int flag=1,cnt=1;
        for(int i=n-1; i>=0; i--) {
            if(cnt%2&&a[i]!=b[i]) {
                if(sum[i]!=i+1-sum[i]) {
                    flag=0;
                    break;
                }
                cnt++;
            }
            else if(cnt%2==0&&a[i]==b[i]) {
                if(sum[i]!=i+1-sum[i]) {
                    flag=0;
                    break;
                }
                cnt++;
            }
        }
        if(flag) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}