gym101964G Matrix Queries seerc2018g题 数学归纳法+线段树（递归）

#include<bits/stdc++.h>
#define CLR(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=(1<<20) +10;
ll sum,tmp;
ll seg[2][maxn<<2],ans[2][21];//ans表示2^k的行（列）有几个同色的 
int k,q;
inline void mode(int id,int o,int l,int r,int x,int dep){
    if(l==r){//染色 
        seg[id][o]^=1;
        return ;
    }
    int mid=(l+r)>>1;
    if(x<=mid){
        mode(id,(o<<1),l,mid,x,dep+1);
    }else{
        mode(id,(o<<1)|1,mid+1,r,x,dep+1);
    }
    if(seg[id][o]!=-1)ans[id][dep]--;  //当成不合法行，先减去 
    if(seg[id][o<<1]==seg[id][(o<<1)|1])seg[id][o]=seg[id][o<<1];//如果左右状态一样 则转移（0,1，-1） 
    else seg[id][o]=-1;
    if(seg[id][o]!=-1)ans[id][dep]++;//仍然合法，加上 
    return ;
}
int main(){
    cin>>k>>q;
    ll n=(1<<k);
    //sum表示总矩阵数  tmp表示合法数 
    for(int i=0;i<k;i++)
    {
        ans[0][i+1]=ans[1][i+1]=1ll*1<<i;
        sum+=1ll<<(i*2);
    }
    
    while(q--)
    {
        int op,x;
        scanf("%d%d",&op,&x);
        
        mode(op,1,1,n,x,1); 
        tmp=0;
        for(int i=0;i<=k;i++)
        {
            tmp+=ans[0][i]*ans[1][i];
        }
        printf("%lld\n",(sum-tmp)*4+1);
    }
}