Codeforces Round #698 (Div. 2)E. Nezzar and Binary String(线段树+离线+异或)

Codeforces Round #698 (Div. 2)E. Nezzar and Binary String(线段树+离线+异或)

题意

给你两个01字符串，要求从第一个字符串转换成第二个字符串，并满足以下要求

每次给定一个检查区间，要求这个区间内必须是单一的0或1

你可以在检查之后偷偷修改严格小于区间一半的字符数

如果可以转换成功输出YES，否则输出NO

思路：

将检查区间，从后往前遍历，则，对于当前检查的区间，之前一定是全0或者全1，所以将当前检查区间和*2与区间长度作比较，若相等则输出NO，若小于则全变为0，大于全变为1

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=3e5+5;
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
int t,n,m,a[maxn],b[maxn],lazy[maxn<<2],sum[maxn<<2];
struct node{
	int l,r;
}p[maxn];
void build(int rt,int l,int r){
    lazy[rt]=-1;
	if(l==r){
		sum[rt]=b[l];
		return;
	}
	int mid=(l+r)>>1;
	build(lson);
	build(rson);
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void pushdown(int rt,int l,int r){
    if(lazy[rt]!=-1){
	    int mid=(l+r)>>1;
	    lazy[rt<<1]=lazy[rt];
	    sum[rt<<1]=(mid-l+1)*lazy[rt];
	    lazy[rt<<1|1]=lazy[rt];
	    sum[rt<<1|1]=(r-mid)*lazy[rt];
        lazy[rt]=-1;
	}
}
void update(int rt,int l,int r,int x,int y,int v){
    if(x<=l&&r<=y){
       lazy[rt]=v;
       sum[rt]=v*(r-l+1);
       return;
    }
    int mid=(l+r)>>1;
    pushdown(rt,l,r);
    if(x<=mid)
      update(lson,x,y,v);
    if(mid<y)
      update(rson,x,y,v);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
int query(int rt,int l,int r,int x,int y){
    if(x<=l&&r<=y)return sum[rt];
    int mid=(l+r)>>1;
	int ans=0;
    pushdown(rt,l,r);
    if(x<=mid)
      ans+=query(lson,x,y);
    if(mid<y)
       ans+=query(rson,x,y);
    return ans;     
}
int main(){
	scanf("%d",&t);
	while(t--){
		scanf("%d%d",&n,&m);
	    for(int i=1;i<=n;i++){
		    scanf("%1d",&a[i]);
	    }
	    for(int i=1;i<=n;i++){
		    scanf("%1d",&b[i]);
	    }
	    build(1,1,n);
	    for(int i=1;i<=m;i++){
		    scanf("%d%d",&p[i].l,&p[i].r);
	    }
	    for(int i=m;i;i--){
		    int he=query(1,1,n,p[i].l,p[i].r),cha=p[i].r-p[i].l+1;
		    if(cha<he*2)update(1,1,n,p[i].l,p[i].r,1);
		    else if(cha>he*2)update(1,1,n,p[i].l,p[i].r,0);
		    else{
			    printf("NO\n");
			    goto abc;
		    }
	    }
	    for(int i=1;i<=n;i++){
			if(a[i]!=query(1,1,n,i,i)){
				printf("NO\n");
				goto abc;
			}
		}
		printf("YES\n");
		abc:;
	}
}