#3688. 毒瘤（duliu）

题目描述

从前有一名毒瘤。

毒瘤最近发现了量产毒瘤题的奥秘。考虑如下类型的数据结构题：给出一个数组，要求支持若干种奇奇怪怪的修改操作（例如给一个区间内的数同时加上 $c$，或者将一个区间内的数同时开平方根），并且支持询问区间的和。毒瘤考虑了 $n$ 个这样的修改操作,并将它们编号为 $1\dots n$。当毒瘤要出数据结构题的时候，他就将这些修改操作中选若干个出来，然后出成一道题。

当然了，这样出的题有可能不可做。通过精妙的数学推理，毒瘤揭露了这些修改操作之间的关系：有 $m$ 对「互相排斥」的修改操作，第 $i$ 对是第 $u_i$ 个操作和第 $v_i$ 个操作。当一道题中同时含有 $u_i$ 和 $v_i$ 这两个操作时，这道题就会变得不可做。另一方面，当一道题中不包含任何「互相排斥」的操作时，这个题就是可做的。此外，毒瘤还发现了一个规律：$m-n$ 是一个很小的数字（参见「数据范围」中的说明），且任意两个修改操作都是连通的。两个修改操作 $a,b$ 是连通的，当且仅当存在若干操作 $t_0,t_1,...,t_l$，使得 $t_0=a,t_l=b$，且对任意 $1\le i\le l$，$t_{i-1}$ 和 $t_i$ 都是「互相排斥」的修改操作。

一对「互相排斥」的修改操作称为互斥对。现在毒瘤想知道，给定值 $n$ 和 $m$ 个互斥对，他一共能出出多少道可做的不同的数据结构题。两个数据结构题是不同的，当且仅当其中某个操作出现在了其中一个题中，但是没有出现在另一个题中。

题解

为什么看到的题解都是虚树啊？这里介绍一个 $\text{ddp}$ 做法。

如果 $m=n-1$ 的话，那就是简单 $\text{dp}$ ： $f[u][0/1]$ 表示 $u$ 这个点选/不选的方案数，转移显然。

那现在如果 $n\le m$ 的话，由于 $m-n+1$ 不大所以我们只要减掉不合法情况即可，容易想到容斥。即枚举状态然后把状态中为 $1$ 的边的两端强制选（即 $f[u][0]=0$ ），然后再做 $\text{dp}$ 即可，可是这样是 $O(2^{m-n+1}n)$ 的过不去。然后我们发现转移可以写成矩阵的形式，于是就可以用树剖+线段树维护矩阵乘即可，效率 $O(2^{m-n+1}lo{g}^{2}n+n)$ 。

代码

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+15,M=N<<1,P=998244353;
int n,m,hd[N],V[M],nx[M],t=1,top[N],sz[N],dp[N],fa[N];
int son[N],f[N][2],h[N],df[N],id[N],b[15],g[N][2],s;
vector<int>G[N][2];bool vis[N];
struct O{int p[2][2];}W,a[N<<2];
int X(int x){return x>=P?x-P:x;}
int K(int x,int y){
	int z=1;
	for (;y;y>>=1,x=1ll*x*x%P)
		if (y&1) z=1ll*z*x%P;
	return z;
}
O Mul(O A,O B){
	for (int i=0;i<2;i++)
		for (int j=0;j<2;j++){
			W.p[i][j]=0;
			for (int k=0;k<2;k++)
				W.p[i][j]=X(W.p[i][j]+1ll*A.p[k][j]*B.p[i][k]%P);
		}
	return W;
}
void add(int u,int v){
	nx[++t]=hd[u];V[hd[u]=t]=v;
}
void dfs(int u,int fr){
	dp[u]=dp[fa[u]=fr]+1;sz[u]=1;vis[u]=1;
	for (int v,i=hd[u];i;i=nx[i]){
		if (vis[v=V[i]]) continue;
		dfs(v,u);sz[u]+=sz[v];
		if (sz[v]>sz[son[u]]) son[u]=v;
	}
}
void Dfs(int u,int tp){
	top[u]=tp;h[u]=df[id[u]=++t]=u;
	if (son[u]) Dfs(son[u],tp),h[u]=h[son[u]];
	for (int i=hd[u];i;i=nx[i])
		if (V[i]!=son[u] && fa[V[i]]==u)
			Dfs(V[i],V[i]);
}
void Dp(int u){
	f[u][0]=f[u][1]=g[u][0]=g[u][1]=1;
	for (int v,i=hd[u];i;i=nx[i])
		if (fa[v=V[i]]==u) Dp(v),
		f[u][0]=1ll*f[u][0]*(f[v][0]+f[v][1])%P,
		f[u][1]=1ll*f[u][1]*f[v][0]%P;
}
#define Ls k<<1
#define Rs k<<1|1
#define mid ((l+r)>>1)
void build(int k,int l,int r){
	if (l==r){
		int u=df[l];
		for (int i=hd[u];i;i=nx[i])
			if (V[i]!=son[u] && fa[V[i]]==u)
				g[u][0]=1ll*g[u][0]*(f[V[i]][0]+f[V[i]][1])%P,
				g[u][1]=1ll*g[u][1]*f[V[i]][0]%P;
		a[k]=(O){g[u][0],g[u][0],g[u][1],0};return;
	}
	build(Ls,l,mid);build(Rs,mid+1,r);
	a[k]=Mul(a[Rs],a[Ls]);
}
O qry(int k,int l,int r,int L,int R){
	if (L<=l && r<=R) return a[k];
	if (mid<L) return qry(Rs,mid+1,r,L,R);
	if (mid>=R) return qry(Ls,l,mid,L,R);
	return Mul(qry(Rs,mid+1,r,L,R),qry(Ls,l,mid,L,R));
}
void upd(int k,int l,int r,int x,int v0,int v1){
	if (l==r){a[k]=(O){v0,v0,v1,0};return;}
	if (mid>=x) upd(Ls,l,mid,x,v0,v1);
	else upd(Rs,mid+1,r,x,v0,v1);
	a[k]=Mul(a[Rs],a[Ls]);
	
}
void del(int x,int o){
	O v;int v0,v1,u;
	while(x!=1){
		v=qry(1,1,n,id[x],id[h[x]]);
		v0=v.p[0][0],v1=v.p[1][0],u=fa[x];
		if (!o) G[u][0].push_back(g[u][0]),G[u][1].push_back(g[u][1]);
		g[u][0]=1ll*g[u][0]*K(X(v0+v1),P-2)%P;
		g[u][1]=1ll*g[u][1]*K(v0,P-2)%P;x=top[u];
	}
}
void ins(int x,int o){
	O v;int v0,v1,u;
	while(x!=1){
		v=qry(1,1,n,id[x],id[h[x]]);
		v0=v.p[0][0],v1=v.p[1][0],u=fa[x];
		if (o){
			int z=G[u][0].size();
			g[u][0]=G[u][0][z-1];G[u][0].pop_back();
			g[u][1]=G[u][1][z-1];G[u][1].pop_back();
		}
		else
			g[u][0]=1ll*g[u][0]*(v0+v1)%P,
			g[u][1]=1ll*g[u][1]*v0%P;
		upd(1,1,n,id[u],g[u][0],g[u][1]);
		x=top[u];
	}
}
int main(){
	cin>>n>>m;
	for (int u,v,i=1;i<=m;i++)
		scanf("%d%d",&u,&v),
		add(u,v),add(v,u);
	t=0;dfs(1,0);Dfs(1,1);
	Dp(1);build(1,1,n);t=0;
	for (int u,v,i=1;i<=m;i++){
		u=V[i<<1],v=V[i<<1|1];
		if (dp[u]>dp[v]) swap(u,v);
		if (fa[v]!=u) b[t++]=i;
	}
	
	for (int F,i=(1<<t)-1;~i;i--){
		F=1;
		for (int j=0;j<t;j++)
			if ((i>>j)&1){
				F=P-F;
				for (int u,k=0;k<2;k++){
					u=V[b[j]<<1|k];del(top[u],0);
					G[u][0].push_back(g[u][0]);
					G[u][1].push_back(g[u][1]);
					upd(1,1,n,id[u],g[u][0]=0,g[u][1]);
					ins(top[u],0);
				}
			}
		O v=qry(1,1,n,1,id[h[1]]);
		s=X(1ll*F*(v.p[0][0]+v.p[1][0])%P+s);
		for (int z,j=t-1;~j;j--)
			if ((i>>j)&1){
				F=P-F;
				for (int u,k=0;k<2;k++){
					u=V[b[j]<<1|k];del(top[u],1);
					z=G[u][0].size();
					g[u][0]=G[u][0][z-1];G[u][0].pop_back();
					g[u][1]=G[u][1][z-1];G[u][1].pop_back();
					upd(1,1,n,id[u],g[u][0],g[u][1]);
					ins(top[u],1);
				}
			}
	}
	cout<<s<<endl;return 0;
}

最后

以上就是痴情跳跳糖为你收集整理的#3688. 毒瘤（duliu）的全部内容，希望文章能够帮你解决#3688. 毒瘤（duliu）所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。