●洛谷P2495 [SDOI2011]消耗战

最新推荐文章于 2025-09-12 17:31:31 发布

weixin_30537391

最新推荐文章于 2025-09-12 17:31:31 发布

阅读量83

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/zj75211/p/8552094.html

本文介绍了一种解决特定图论问题的方法——利用虚树的概念进行树形动态规划。通过对给定节点集构建虚树，并运用LCA算法辅助，实现了最小路径值的高效计算。

题链：

https://www.luogu.org/problemnew/show/P2495
题解：

虚树入门，树形dp
推荐博客：http://blog.youkuaiyun.com/lych_cys/article/details/50814948

代码：

#include<bits/stdc++.h>
#define MAXN 250005
#define INFll 0x3f3f3f3f3f3f3f3fll
using namespace std;
int N,Q,M;
bool mark[MAXN];
long long val[MAXN];
int mincut[MAXN],dfn[MAXN],deep[MAXN],fa[MAXN][19];
struct Edge{
	int ent;
	int to[MAXN*2],val[MAXN*2],nxt[MAXN*2],head[MAXN];
	Edge(){ent=2;}
	void Adde(int u,int v,int w){
		to[ent]=v; val[ent]=w;
		nxt[ent]=head[u]; head[u]=ent++;
	}
}E1,E2;
bool cmp(int a,int b){return dfn[a]<dfn[b];}
void read(int &x){
	static int sign; static char ch;
	sign=1; x=0; ch=getchar();
	for(;ch<'0'||'9'<ch;ch=getchar()) if(ch=='-') sign=-1;
	for(;'0'<=ch&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	if(sign==-1) x=-x;
}
void dfs(int u,int dep){
	static int dnt;
	dfn[u]=++dnt; deep[u]=dep;
	for(int k=1;k<19;k++)
		fa[u][k]=fa[fa[u][k-1]][k-1];
	for(int e=E1.head[u];e;e=E1.nxt[e]){
		int v=E1.to[e]; if(v==fa[u][0]) continue;
		fa[v][0]=u; val[v]=min(val[u],1ll*E1.val[e]);
		dfs(v,dep+1);
	}
}
int LCA(int u,int v){
	if(deep[u]>deep[v]) swap(u,v);
	for(int k=18;k>=0;k--) if(deep[fa[v][k]]>=deep[u]) v=fa[v][k];
	if(u==v) return u;
	for(int k=18;k>=0;k--) if(fa[u][k]!=fa[v][k]) u=fa[u][k],v=fa[v][k];
	return fa[u][0];
}
long long dp(int u){
	if(!E2.head[u]) return val[u];
	long long tmp=0;
	for(int e=E2.head[u];e;e=E2.nxt[e]){
		int v=E2.to[e]; tmp+=dp(v);
	}
	E2.head[u]=0; 
	if(mark[u]) return val[u];
	return min(tmp,val[u]);
}
void solve(){
	static int a[MAXN*2],stk[MAXN*2],top,lca;
	read(M); E2.ent=2; top=0;
	for(int i=1;i<=M;i++) read(a[i]),mark[a[i]]=1;
	//虚树的构建------------------------------------
	sort(a+1,a+M+1,cmp); stk[++top]=1;
	for(int i=1;i<=M;i++){
		lca=LCA(stk[top],a[i]);
		if(lca!=stk[top]) while(1){
			if(dfn[stk[top-1]]<=dfn[lca]){
				E2.Adde(lca,stk[top],0),top--;
				if(stk[top]!=lca) stk[++top]=lca;
				break;
			}
			E2.Adde(stk[top-1],stk[top],0),top--;
		}
		if(stk[top]!=a[i]) stk[++top]=a[i];
	}
	while(top>1) E2.Adde(stk[top-1],stk[top],0),top--;
	//-----------------------------------------------
	printf("%lld\n",dp(1));
	for(int i=1;i<=M;i++) mark[a[i]]=0;
}
int main(){
	read(N);
	for(int i=1,a,b,c;i<N;i++){
		read(a),read(b),read(c);
		E1.Adde(a,b,c); E1.Adde(b,a,c);
	}
	val[1]=INFll; dfs(1,1); read(Q);
	while(Q--) solve();
	return 0;
}

转载于:https://www.cnblogs.com/zj75211/p/8552094.html