bzoj4543: [POI2014]Hotel加强版

本文详细介绍了一种结合深度优先搜索(DFS)和动态规划(DP)的算法,用于解决特定类型的图论问题。该算法首先通过DFS遍历图结构,然后使用DP优化路径选择,以求得全局最优解。文章提供了完整的C++代码实现,并解释了关键步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目
题解

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100002;
struct node{
	int to,ne;
}e[N<<1];
int n,i,x,y,tot,dep[N],h[N],son[N];
ll space[N*7],*f[N],*g[N],ans,*now=space+N;
inline char gc(){
	static char buf[100000],*p1=buf,*p2=buf;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int rd(){
	int x=0,fl=1;char ch=gc();
	for (;ch<48||ch>57;ch=gc())if(ch=='-')fl=-1;
	for (;48<=ch&&ch<=57;ch=gc())x=(x<<3)+(x<<1)+(ch^48);
	return x*fl;
}
inline void wri(int a){if(a<0)a=-a,putchar('-');if(a>=10)wri(a/10);putchar(a%10|48);}
inline void wln(int a){wri(a);puts("");}
void add(int x,int y){e[++tot]=(node){y,h[x]},h[x]=tot;}
void create(int u){
	f[u]=now,now+=dep[u]*2+1;
	g[u]=now,now+=dep[u]*2+1;
}
void init(int u,int fa){
	for (int i=h[u],v;i;i=e[i].ne)
		if ((v=e[i].to)!=fa){
			init(v,u);
			if (dep[v]>dep[son[u]]) son[u]=v;
		}
	dep[u]=dep[son[u]]+1;
}
void dfs(int u,int fa){
	f[u][0]=1;
	if (son[u]){
		f[son[u]]=f[u]+1;
		g[son[u]]=g[u]-1;
		dfs(son[u],u);
		ans+=g[u][0];
	}
	for (int i=h[u],v;i;i=e[i].ne)
		if ((v=e[i].to)!=fa && v!=son[u]){
			create(v);
			dfs(v,u);
			for (int j=dep[v];~j;j--){
				if (j) ans+=f[u][j-1]*g[v][j];
				ans+=g[u][j+1]*f[v][j];
				g[u][j+1]+=f[u][j+1]*f[v][j];
			}
			for (int j=0;j<=dep[v];j++){
				if (j) g[u][j-1]+=g[v][j];
				f[u][j+1]+=f[v][j];
			}
		}
}
int main(){
	n=rd();
	for (i=1;i<n;i++) x=rd(),y=rd(),add(x,y),add(y,x);
	init(1,0);
	create(1);
	dfs(1,0);
	printf("%lld",ans);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值