[ZJOI2019]线段树

ZJOI2019最水的一道题,考场上想出了正解却死活调不出来。。

然后就一直咕到了现在(逃

题目求的那个东西可以转化为这样一个东西。

有一些操作,你可以决定每个操作是否执行,动态查询所有可能的情况中线段树上权值为1的节点个数的总和。

然后很自然的(虽然看起来有点奇怪,但真的就很自然的能想到!)想到把维护方案数变成维护期望。

然后根据期望的线性性,等价于维护每个点权值为1的概率。

这样的好处是,如果维护方案数的话,那些不受影响的节点每次操作需要乘上2,维护概率的话则不需要改变。

然后经过一波毒瘤的分类套路,你大概需要设计这样一个dp状态,dp[o][0/1][0/1]表示o这个点,是否为0/1,它到根的路径是否有0/1的概率。

转移方程很普及组,就是有点繁琐。

然后你可以把所有点按照不同的转移方式划分为4类。

1.optset时会被遍历到的点。

2.不会被遍历到但却会被pushdown到的点

3.遍历到的点中最底端,也就是ql<=l&&r<=qr的点

4.第3种点的子树

前三种都可以直接修改,第四种可以通过打标记实现修改。

#include<bits/stdc++.h>
#define N 220000
#define eps 1e-7
#define inf 1e9+7
#define db double
#define ll long long
#define ldb long double
using namespace std;
inline int read()
{
    char ch=0;
    int x=0,flag=1;
    while(!isdigit(ch)){ch=getchar();if(ch=='-')flag=-1;}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*flag;
}
const int mo=998244353;
int ksm(int x,int k)
{
    int ans=1;
    while(k){if(k&1)ans=1ll*ans*x%mo;k>>=1;x=1ll*x*x%mo;}
    return ans;
}
int inv(int x){return ksm(x,mo-2);}
int v,mi[N],mv[N];
struct Segment_Tree
{
    #define lson o<<1
    #define rson o<<1|1
    #define mid ((l+r)>>1)
    int lx[N*4],rx[N*4];
    int sumv[N*4],tagv[N*4],f[N*4][2][2],g[N*4][2][2];
    inline void pushup(int o)
    {
        sumv[o]=(sumv[lson]+sumv[rson])%mo;
        sumv[o]=(sumv[o]+((f[o][1][0]+f[o][1][1])%mo))%mo;
    }
    inline void update(int o,int t)
    {
    	int k=mv[t];tagv[o]+=t;
        for(int i=0;i<=1;i++)
        {
            f[o][i][1]=(f[o][i][1]+(1ll*f[o][i][0]*(1-k)%mo))%mo;
            f[o][i][0]=1ll*f[o][i][0]*k%mo;
        }
    }
    inline void pushdown(int o)
    {
        update(lson,tagv[o]);
		update(rson,tagv[o]);
		tagv[o]=0;
    }
    void build(int o,int l,int r)
    {
    	lx[o]=l;rx[o]=r;
        f[o][0][0]=1;
        if(l==r)return;
        build(lson,l,mid);build(rson,mid+1,r);
    }
    void print(int o)
    {
    	cout<<lx[o]<<"--"<<rx[o]<<endl;
    	for(int i=0;i<=1;i++)for(int j=0;j<=1;j++)
    	cout<<(f[o][i][j]%mo+mo)%mo<<" ";cout<<endl<<sumv[o]<<endl<<endl;
    }
    void copy(int o)
    {
    	for(int i=0;i<=1;i++)
		for(int j=0;j<=1;j++)
		g[o][i][j]=f[o][i][j];
    }
    void solve(int o,int l,int r,int ql,int qr)
	{
		copy(o);
		if(ql<=l&&r<=qr)return;
		pushdown(o);
		if(ql<=mid)solve(lson,l,mid,ql,qr);else copy(lson);
		if(qr>mid)solve(rson,mid+1,r,ql,qr);else copy(rson);
	}
	void opt(int o,int l,int r)//被pushdown到的节点 
    {
    	f[o][0][0]=g[o][0][0];
    	f[o][0][1]=1ll*v*g[o][0][1]%mo;
    	f[o][1][0]=(g[o][1][0]+(1ll*(g[o][0][1]+g[o][1][1])%mo*v%mo))%mo;
    	f[o][1][1]=1ll*v*g[o][1][1]%mo;
    	sumv[o]=(f[o][1][0]+f[o][1][1])%mo;
    	if(l!=r)sumv[o]=(sumv[o]+((sumv[lson]+sumv[rson])%mo))%mo;
    }
    void optset(int o,int l,int r,int ql,int qr)
    {
    	if(ql<=l&&r<=qr)
    	{
    		//边界点
    		f[o][0][0]=1ll*v*g[o][0][0]%mo;
    		f[o][0][1]=1ll*v*g[o][0][1]%mo;
    		f[o][1][0]=1ll*v*(1+g[o][1][0])%mo;
    		f[o][1][1]=1ll*v*g[o][1][1]%mo;
    		sumv[o]=(f[o][1][0]+f[o][1][1])%mo;
			if(l!=r)//边界点的子树
			{
				update(lson,1),update(rson,1);
				sumv[o]=(sumv[o]+((sumv[lson]+sumv[rson])%mo))%mo;
			}
			return;
    	}
    	if(ql<=mid)optset(lson,l,mid,ql,qr);else opt(lson,l,mid);
		if(qr>mid)optset(rson,mid+1,r,ql,qr);else opt(rson,mid+1,r);
		//被访问到的点 
		f[o][0][0]=1ll*v*(1+g[o][0][0])%mo;
		f[o][0][1]=1ll*v*g[o][0][1]%mo;
		f[o][1][0]=1ll*v*g[o][1][0]%mo;
		f[o][1][1]=1ll*v*g[o][1][1]%mo;
		pushup(o);
    }
    int query(){return sumv[1];}
}T;
int main()
{
    int n=read(),m=read();
    v=inv(2);mi[0]=mv[0]=1;T.build(1,1,n);
    for(int i=1;i<=n+m;i++)
    {
    	mi[i]=1ll*mi[i-1]*2%mo;
    	mv[i]=1ll*mv[i-1]*v%mo;
    }
	for(int i=1,cnt=0;i<=m;i++)
    {
        int flag=read();
        if(flag==1)
        {
        	cnt++;
            int l=read(),r=read();
            T.solve(1,1,n,l,r);
            T.optset(1,1,n,l,r);
        }
        if(flag==2)
        {
            int ans=1ll*T.query()*mi[cnt]%mo;
            printf("%d\n",(ans%mo+mo)%mo);
        }
    }
    return 0;
}
posted @ 2019-04-27 22:57  Creed-qwq  阅读(127)  评论(0编辑  收藏  举报