BZOJ 2286:Sdoi2011消耗战

10 篇文章 0 订阅
2 篇文章 0 订阅
也是新学虚树,所以只能简单介绍下。
想必大家都知道离散化,虚树可以理解为树上的离散化,这样处理,对于一些询问次数比较多,原树比较大的题,我们就可以尝试建立出虚树,并在虚树上进行操作。
而虚树作为一个工具,一般会和树形dp或树上贪心结合在一起。
很幸运看了yyb的博客后,发现了很简便的建虚树的方法,此题建好虚树后,一个基础树形dp即可。
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=3e5+5;
int n,u,v,w,q,k;
int tot,dfn[N],low[N],size[N],d[N],f[N],val[N],son[N],top[N];
int p[N<<1],sta[N];
bool vis[N];
int cnt,head[N];
struct edge{int next,to,w;}e[N<<1];

inline void add(int u,int v,int w)
{
	cnt++;
	e[cnt].next=head[u];
	e[cnt].to=v;
	e[cnt].w=w;
	head[u]=cnt;
}

void dfs(int u,int fa)
{
	dfn[u]=++tot;
	size[u]=1;
	for (register int i=head[u]; i; i=e[i].next)
	if (e[i].to!=fa)
	{
		d[e[i].to]=d[u]+1; f[e[i].to]=u;
		val[e[i].to]=min(val[u],e[i].w);
		dfs(e[i].to,u);
		size[u]+=size[e[i].to];
		if (size[son[u]]<size[e[i].to]) son[u]=e[i].to;
	}
	low[u]=tot;
}

void dfs2(int u,int TP)
{
	top[u]=TP;
	if (son[u]) dfs2(son[u],TP);
	for (register int i=head[u]; i; i=e[i].next)
	if (e[i].to!=son[u] && e[i].to!=f[u]) dfs2(e[i].to,e[i].to);
}

inline int lca(int u,int v)
{
	while (top[u]!=top[v])
	{
		if (d[top[u]]<d[top[v]]) swap(u,v);
		u=f[top[u]];
	}
	if (d[u]>d[v]) swap(u,v);
	return u;
}

int DP(int u)
{
	if (vis[u]) return val[u];
	int now=0;
	for (register int i=head[u]; i; i=e[i].next) now+=DP(e[i].to);
	return min(now,val[u]);
}

inline bool cmp(int a,int b){return dfn[a]<dfn[b];}

signed main(){
	scanf("%lld",&n);
	for (register int i=1; i<n; ++i)
	{
		scanf("%lld%lld%lld",&u,&v,&w);
		add(u,v,w); add(v,u,w);	
	}
	val[1]=1e15;
	dfs(1,0); dfs2(1,1);
	memset(head,0,sizeof(head));
	scanf("%lld",&q);
	while (q--)
	{
		cnt=0;
		scanf("%lld",&k);
		for (register int i=1; i<=k; ++i) scanf("%lld",&p[i]),vis[p[i]]=true;
		sort(p+1,p+k+1,cmp);
		for (register int i=k; i>1; --i) p[++k]=lca(p[i],p[i-1]); 
		sort(p+1,p+k+1,cmp);
		k=unique(p+1,p+k+1)-p-1;
		for (register int i=1,top=0; i<=k; ++i)
		{
			while (top && low[sta[top]]<dfn[p[i]]) top--;
			add(sta[top],p[i],0); sta[++top]=p[i];
		}
		printf("%lld\n",DP(p[1]));
		for (register int i=1; i<=k; ++i) head[p[i]]=0,vis[p[i]]=false;
	}
return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值