P2495 [SDOI2011]消耗战

最新推荐文章于 2024-10-03 17:28:04 发布

原创最新推荐文章于 2024-10-03 17:28:04 发布 · 323 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构-虚树 #算法-LCA #DP-树形DP

算法-LCA 同时被 3 个专栏收录

3 篇文章

订阅专栏

DP-树形DP

3 篇文章

订阅专栏

数据结构-虚树

1 篇文章

订阅专栏

本文详细解析了SDOI2011消耗战问题，通过构建虚树并运用动态规划解决关键点的最小花费问题。介绍了虚树的构建方法，以及如何通过DFS遍历和LCA查询优化算法。

P2495 [SDOI2011]消耗战

题目描述

详见：P2495 [SDOI2011]消耗战

Solution

此题是虚树的经典版子题吧qwq。

是不是直接贴代码就行了啊（反正不是luogu题解，没有人查）。

大概就是先建出虚树（一般给定一堆关键点的题都是虚树），然后DP。

设 $f[x]$ 为处理掉 $x$ 子树中所有关键点的最小花费， $mn[x]$ 为从 $x$ 到根的路径上的最小边权。

$f[x]=min(mn[x],\sum f[son])$

然后……没了。

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=5e5+50; 
const ll INF=1ll<<60;
int Log[MAXN],fa[MAXN][21],dfn[MAXN],dep[MAXN];
int stk[MAXN],a[MAXN],top,DFN=0;
ll mn[MAXN],f[MAXN];
struct enode{int to; ll c; };
vector<enode> e[MAXN];
vector<int> son[MAXN];

inline int read()
{
    int x=0,f=1; char c=getchar();
    while (c<'0'||c>'9') { if(c=='-') f=-1; c=getchar(); }
    while (c>='0'&&c<='9') { x=(x<<3)+(x<<1)+(c^48); c=getchar(); }
    return x*f;
}
void add_edge(int u,int v){ son[u].push_back(v); } 
void dfs(int x,int father)
{
	fa[x][0]=father;
	dep[x]=dep[father]+1;
	dfn[x]=++DFN;
	for (int i=1;i<=Log[dep[x]];i++) fa[x][i]=fa[fa[x][i-1]][i-1];
	for (int i=0;i<e[x].size();i++)
	    if (e[x][i].to!=father)
	    {
	    	mn[e[x][i].to]=min(mn[x],e[x][i].c);
	    	dfs(e[x][i].to,x);
	    }
}
int get_lca(int x,int y)
{
	if (dep[x]<dep[y]) swap(x,y);
	for (int i=Log[dep[x]];i>=0;i--)
	    if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
	if (x==y) return x;
	for (int i=Log[dep[x]];i>=0;i--)
	    if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
	return fa[x][0];
}
void insert(int x)
{
	if (top==1) { stk[++top]=x; return; }
	int lca=get_lca(stk[top],x);
	if (lca==stk[top]) return;
	while (top>1&&dfn[stk[top-1]]>=dfn[lca]) add_edge(stk[top-1],stk[top]),top--;
	if (lca!=stk[top]) add_edge(lca,stk[top]),stk[top]=lca;
	stk[++top]=x;
}
ll tree_dp(int x)
{
	for (int i=0;i<son[x].size();i++) tree_dp(son[x][i]);
	f[x]=mn[x];
	if (!son[x].size()) return f[x];
	ll s=0;
	for (int i=0;i<son[x].size();i++) s+=f[son[x][i]];
	son[x].clear();
	return f[x]=min(f[x],s);
}
int compare(int x,int y){ return dfn[x]<dfn[y]; }
int main()
{
	int n=read();
	for (int i=1;i<n;i++)
	{
		int u=read(),v=read(),c=read();
		e[u].push_back((enode){v,c});
		e[v].push_back((enode){u,c});
	}
	dep[0]=-1,Log[1]=0;
	for (int i=2;i<=n;i++) Log[i]=Log[i>>1]+1;
	for (int i=1;i<=n;i++) mn[i]=INF; 
	dfs(1,0);
    //for (int i=1;i<=n;i++) cout<<i<<":"<<dep[i]<<" "<<dfn[i]<<" "<<mn[i]<<endl;
	
	int Case=read();
	while (Case--)
	{
		int m=read();
		for (int i=1;i<=m;i++) a[i]=read();
		sort(a+1,a+m+1,compare);
		stk[top=1]=1;
		for (int i=1;i<=m;i++) insert(a[i]);
		while (top) add_edge(stk[top-1],stk[top]),top--;
		printf("%lld\n",tree_dp(1));
	}
	return 0;
}