CF-1252K-Addition Robot（线段树合并）

题目链接：https://codeforces.com/contest/1252/problem/K

题目大意：给出一个AB串，给出一个算法。q次操作，1：反转，将区间[l,r]中的A变为B，B变为A。2.查询，每次给出查询区间[l,r]，a，b计算最后a和b。

思路：题意很明显可以转换成求区间[l,r]中有多少a和b。我们可以进行合并的时候维护。当时打比赛的时候，队友直接推出了合并公式，类似矩阵乘法一样的东西。区间反转的时候两两交换一下。标记一个lazy标记。

我只是一个没有感情的代码手，照着合并公式写就好了。

ACCode：

#include<stdlib.h>
#include<string.h>
#include<stdio.h>
#include<time.h>
#include<math.h>
// srand((unsigned)time(NULL));rand();
      
#include<map>//unordered_map
#include<set>//multiset
#include<deque>
#include<queue>
#include<stack>
#include<bitset>
#include<string>
#include<fstream>
#include<iostream>
#include<algorithm>
 
#define ll long long
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define clean(a,b) memset(a,b,sizeof(a))
using namespace std;
      
const int MAXN=1e5+10;
//const int MAXM=10;
const int INF32=0x3f3f3f3f;
const ll INF64=0x3f3f3f3f3f3f3f3f;
const ll MOD=1e9+7;
const double PI=acos(-1.0);
const double EPS=1.0e-8;
//unsigned register
// ios::sync_with_stdio(false)

struct SegTree{
	struct Tree{
		int l,r;
		ll Aa,Ab,Ba,Bb;
		int lazy;
		friend Tree operator + (Tree a,Tree b){
			Tree tmp;
			tmp.Aa=(b.Aa*a.Aa)%MOD+(b.Ab*a.Ba)%MOD;
			tmp.Ab=(b.Aa*a.Ab)%MOD+(b.Ab*a.Bb)%MOD;
			tmp.Ba=(b.Ba*a.Aa)%MOD+(b.Bb*a.Ba)%MOD;
			tmp.Bb=(b.Ba*a.Ab)%MOD+(b.Bb*a.Bb)%MOD;
			return tmp;
		}
	};
	Tree T[MAXN<<2];
	
	void PushUp(int rt){
		int l=T[rt].l,r=T[rt].r,lazy=T[rt].lazy;
		T[rt]=T[rt<<1]+T[rt<<1|1];
		T[rt].l=l;T[rt].r=r;T[rt].lazy=lazy;
	}
	void Down(int rt){
		swap(T[rt].Aa,T[rt].Ab);swap(T[rt].Ba,T[rt].Bb);
		swap(T[rt].Aa,T[rt].Ba);swap(T[rt].Ab,T[rt].Bb);
	}
	void PushDown(int rt){
//		printf("lazy=%d\n",T[rt].lazy);
		if(T[rt].lazy!=0){
			Down(rt<<1);T[rt<<1].lazy^=1;
			Down(rt<<1|1);T[rt<<1|1].lazy^=1;
			T[rt].lazy^=1;
		}
	}
	void Build(int l,int r,int rt,char S[]){
		T[rt].l=l;T[rt].r=r;
		T[rt].lazy=0;
		if(l==r){
			if(S[l]=='A'){
				T[rt].Aa=T[rt].Ab=1;
				T[rt].Ba=0;T[rt].Bb=1;
			}
			else{
				T[rt].Aa=1;T[rt].Ab=0;
				T[rt].Ba=T[rt].Bb=1;
			}return ;
		}int mid=(l+r)>>1;
		Build(l,mid,rt<<1,S);Build(mid+1,r,rt<<1|1,S);
		PushUp(rt);
	}
	void Update(int ql,int qr,int rt){
		if(ql<=T[rt].l&&T[rt].r<=qr){
			Down(rt);T[rt].lazy^=1;
			return ;
		}PushDown(rt);
		if(qr<=T[rt<<1].r) Update(ql,qr,rt<<1);
		else if(ql>=T[rt<<1|1].l) Update(ql,qr,rt<<1|1);
		else{
			Update(ql,qr,rt<<1);
			Update(ql,qr,rt<<1|1);
		}PushUp(rt);
	}
	Tree Query(int ql,int qr,int rt){
		if(ql<=T[rt].l&&T[rt].r<=qr){
			return T[rt];
		}PushDown(rt);
		if(qr<=T[rt<<1].r) return Query(ql,qr,rt<<1);
		else if(ql>=T[rt<<1|1].l) return Query(ql,qr,rt<<1|1);
		else return Query(ql,qr,rt<<1)+Query(ql,qr,rt<<1|1);
	}
	void Show(int rt){
		printf("l=%d r=%d A(%lld,%lld) B(%lld,%lld)\n",T[rt].l,T[rt].r,T[rt].Aa,T[rt].Ab,T[rt].Ba,T[rt].Bb);
		if(T[rt].l==T[rt].r) return ;
		PushDown(rt);
		Show(rt<<1);Show(rt<<1|1);
	}
};
SegTree Seg;
char S[MAXN];
int n,m;

int main(){
	while(~scanf("%d%d",&n,&m)){
		scanf("%s",S+1);
		Seg.Build(1,n,1,S);
		while(m--){
			int opt;scanf("%d",&opt);
			if(opt==1){//区间反转 
				int l,r;scanf("%d%d",&l,&r);
				Seg.Update(l,r,1);
			}
			else{//区间查询 
				int tmp=(MAXN<<2)-2;
				int l,r,a,b;scanf("%d%d%d%d",&l,&r,&a,&b);
				Seg.T[tmp]=Seg.Query(l,r,1);
				ll ansa=((1ll*Seg.T[tmp].Aa*a)%MOD+(1ll*Seg.T[tmp].Ab*b)%MOD)%MOD;
				ll ansb=((1ll*Seg.T[tmp].Ba*a)%MOD+(1ll*Seg.T[tmp].Bb*b)%MOD)%MOD;
				printf("%lld %lld\n",ansa,ansb);
			}
//			Seg.Show(1);
		}
	}
}