2023 CCPC 华为云计算挑战赛 hdu7401 流量监控(树形dp)

题目

流量监控 - HDU 7401 - Virtual Judge

简单来说,T(T<=20)组样例,sumn不超过2e4

每次给定一棵n(n<=2000)个点的树,两问:

①将n个点恰拆成n/2个pair(u,v),要求一个点是另一个点的祖先,求方案数

②若两个pair(u,v)、(w,x)满足:

u是v、w、x的祖先,w是v、x的祖先,v是x的祖先

即,四个点都在x通往根的路径上,且[u,v]和[w,x]相交,则称形成了一个区间交,

在①的所有合法方案数中,求区间交的总数

输出①、②的值,答案对998244353取模

思路来源

jiangly代码&heltion&tiger2005&夏老师

题解

对着jiangly代码,找了若干人讨论,终于讨论明白了

第一问,dp[i][j]表示i子树内当前有j个未匹配的点的方案数,

转移是一个树上背包,对子树做完树上背包之后,

再考虑u这个点的决策,要么是(,要么是)

换句话说,要么选一个之前未匹配的点进行匹配,要么新增一个未匹配的点

第二问,长度为4的祖先链(都在通往祖先的一条路径上),所以可以考虑把0-4都维护上,

这里实际是考虑每个长度为4的链的贡献,

即在dp的时候并不指定这四个点连的方式,只统计四元组的总方案数,

然后根据题目要求, 最后的时候将13相连、24相连

这相当于求从树上抠掉四个点(四个点在一条祖先链上)时,剩下的点构成合法方案的方案数

f[i][j][k]表示考虑到j的子树,当前抠掉了i个点,还有k个点没有匹配的方案数

相当于一个二维背包,i是一维,k是一维

转移先对u的子树v1、v2、...做背包,做k这一维的背包,

又因为不同子树之间的点并不在一条祖先链上,

所以i这一维做背包两两合并的时候,两棵子树的i这一维不能同时大于0

cpp 复制代码
        rep(a,0,4){
			rep(b,0,4){
				if(a && b)continue;
				rep(i,0,sz[u]){
					rep(j,0,sz[v]){
						add(tmp[a+b][i+j],1ll*f[a][u][i]*f[b][v][j]%mod);
					}
				}
			}
		}

将子树都合并进来之后,再考虑u的决策,

u的决策实际有三种, 要么是(,要么是),要么从树上抠掉

代码

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
typedef long long ll;
typedef double db;
typedef pair<ll,ll> P;
#define fi first
#define se second
#define pb push_back
#define dbg(x) cerr<<(#x)<<":"<<x<<" ";
#define dbg2(x) cerr<<(#x)<<":"<<x<<endl;
#define SZ(a) (int)(a.size())
#define sci(a) scanf("%d",&(a))
#define pt(a) printf("%d",a);
#define pte(a) printf("%d\n",a)
#define ptlle(a) printf("%lld\n",a)
#define debug(...) fprintf(stderr, __VA_ARGS__)
const int N=2e3+10,mod=998244353,inv2=(mod+1)/2;
int T,n,u,v,sz[N],tmp[5][N],f[5][N][N];
vector<int>e[N];
void add(int &x,int y){
	x=(x+y)%mod;
}
void dfs(int u,int fa){
	sz[u]=1;
	f[0][u][0]=1;
	for(auto &v:e[u]){
		if(v==fa)continue;
		dfs(v,u);
		memset(tmp,0,sizeof tmp);
		rep(a,0,4){
			rep(b,0,4){
				if(a && b)continue;
				rep(i,0,sz[u]){
					rep(j,0,sz[v]){
						add(tmp[a+b][i+j],1ll*f[a][u][i]*f[b][v][j]%mod);
					}
				}
			}
		}
		sz[u]+=sz[v];
		rep(a,0,4){
			rep(i,0,sz[u]){
				f[a][u][i]=tmp[a][i];
			}
		}
	}
	memset(tmp,0,sizeof tmp);
	rep(a,0,4){
		rep(i,0,sz[u]){
			if(i)add(tmp[a][i-1],1ll*f[a][u][i]*i%mod);
			add(tmp[a][i+1],f[a][u][i]);
			if(a<4)add(tmp[a+1][i],f[a][u][i]);
		}
	}
	rep(a,0,4){
		rep(i,0,sz[u]){
			f[a][u][i]=tmp[a][i];
		}
	}
}
int main(){
	scanf("%d",&T);
	while(T--){
		sci(n);
		rep(i,1,n){
			e[i].clear();
			sz[i]=0;
		}
		memset(f,0,sizeof f);
		rep(i,2,n){
			sci(u),sci(v);
			e[u].pb(v);
			e[v].pb(u);
		}
		dfs(1,0);
		printf("%d %d\n",f[0][1][0],f[4][1][0]);
	}
	return 0;
}
相关推荐
陈琦煜2 天前
AcWing 1073 树的中心 树形dp (详解)
动态规划·树形dp
呆呆敲代码的小Y16 天前
【Unity 实用工具篇】 | UGUI 循环列表 SuperScrollView,快速上手使用
游戏·unity·游戏引擎·背包·循环组件
cocoack8 个月前
备战蓝桥杯---树形DP基础2
c++·算法·蓝桥杯·深度优先·树形dp
卡布叻_周深9 个月前
2024初三年后集训模拟测试3
dp·树形dp·二分答案·单调队列优化dp
闻缺陷则喜何志丹9 个月前
【动态规划】【树形dp】【C++算法】968监控二叉树
c++·算法·动态规划·力扣·监控·树形dp·摄像头
Dugege0079 个月前
【Unity】QFramework通用背包系统优化:使用Odin优化编辑器
unity·编辑器·背包·qframework·通用·odin
Qres8211 年前
可删除背包(计数类): P4141
背包
源代码•宸1 年前
Leetcode—2646.最小化旅行的价格总和【困难】
c++·经验分享·算法·leetcode·深度优先·树形dp
小威W1 年前
【力扣周赛】第 115 场双周赛(⭐优化背包DP)(TODO)
java·算法·leetcode·力扣·周赛·dp·背包
Landing_on_Mars1 年前
K. Kingdom‘s Power,树形dp
算法·深度优先·图论·树形dp