[ARC121F]Logical Operations on Tree

Logical Operations on Tree

题解

简单dp

首先我们很容易发现一种贪心的手段。
我们可以通过操作使树的一个叶子的值为$1$，并它连向其父亲的边是$or$，我们就将这个叶子保留到最后。
否则我们再枚举完这个点的子树后就向上操作一定是最优的。

• 当叶子为$1$，并且边是$or$时，如果我们保留下来最后再来$or1$，则最后剩下来的节点一定是$1$，我们会对其进行单独计算。
• 当叶子为$1$，并且边是$and$时，无论我们什么时候向上进行都不会对答案造成影响，还不如直接现在进行。
• 当叶子为$0$，并且边是$or$时，也不会对答案造成影响，也可以直接现在进行。
• 当叶子为$0$，并且边是$and$时，由于只有$or1$会使该节点的值改变为$1$，但$or1$我们是对其进行单独计算的，所以这个点的值为$1$时当且仅当它本来就是$1$，这个$and0$一定会对其造成影响，不如就现在进行。而这个点值为$0$时$and0$根本不会产生任何影响，不如就现在进行。

于是就可以证明我们的贪心做法是正确的，接下来可以用$dp$准确说并不是dp，只是一个计数对树的形态进行计算。

令$f_{x,0/1}$表示$x$节点的子树中不含$or1$，且它们全部操作完后得到的节点值为$0/1$的树的个数。
令$g_x$表示$x$的子树中含有$or1$的方案数，我们的每次操作相当于对两棵子树进行合并，
容易得到状态转移方程式：
$$f_{u,0}=2f_{u,0}{f}_{v,0}+f_{u,0}{f}_{v,1}+f_{u,1}{f}_{v,0}$$
代表$0or0,0and0,0and1,1and0$四种情况。
$$f_{u,0}=f_{u,1}(f_{v,0}+f_{v,1})$$
代表$1or0,1and1$四种情况。
$$g_u=2g_u(f_{v,0}+f_{v,1}+g_v)+2(f_{u,0}+f_{u,1})g_v+(f_{u,0}+f_{u,1})f_{v,1}$$
代表$u$或$v$子树中已有$or1$或$0or1,1or1$的情况。
答案就是$f_{1,1}+g_1$。

时间复杂度$O\left(n\right)$。

源码

#include<bits/stdc++.h>
using namespace std;
#define MAXN 100005
#define lowbit(x) (x&-x)
#define reg register
#define mkpr make_pair
#define fir first
#define sec second
typedef long long LL;
typedef unsigned long long uLL;
const int INF=0x3f3f3f3f;
const int mo=998244353;
const int zero=500;
const LL jzm=2333;
const int orG=3,invG=332748118;
const double Pi=acos(-1.0);
typedef pair<int,int> pii;
const double PI=acos(-1.0);
template<typename _T>
_T Fabs(_T x){return x<0?-x:x;}
template<typename _T>
void read(_T &x){
	_T f=1;x=0;char s=getchar();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=getchar();}
	while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
	x*=f;
}
template<typename _T>
void print(_T x){if(x<0){x=(~x)+1;putchar('-');}if(x>9)print(x/10);putchar(x%10+'0');}
int add(int x,int y){return x+y<mo?x+y:x+y-mo;}
int dp[MAXN][2],g[MAXN],n,head[MAXN],tot;
struct edge{int to,nxt;}e[MAXN<<1];
void addEdge(int u,int v){e[++tot]=(edge){v,head[u]};head[u]=tot;}
void dosaka(int u,int fa){
	dp[u][0]=dp[u][1]=1;
	for(int i=head[u];i;i=e[i].nxt){
		int v=e[i].to;if(v==fa)continue;dosaka(v,u);
		int t1=dp[u][0],t2=dp[u][1],tmp=add(dp[v][0],dp[v][1]);
		dp[u][0]=add(add(2ll*t1*dp[v][0]%mo,1ll*t1*dp[v][1]%mo),1ll*t2*dp[v][0]%mo);
		dp[u][1]=add(1ll*t2*dp[v][0]%mo,1ll*t2*dp[v][1]%mo);
		g[u]=add(2ll*g[u]*add(g[v],tmp)%mo,2ll*add(t1,t2)*g[v]%mo);
		g[u]=add(g[u],1ll*add(t1,t2)*dp[v][1]%mo);
	}
}
signed main(){
	read(n);
	for(int i=1;i<n;i++){
		int u,v;read(u);read(v);
		addEdge(u,v);addEdge(v,u);
	}
	dosaka(1,0);
	printf("%d\n",add(g[1],dp[1][1]));
	return 0;
}

谢谢！！！