BZOJ 2286: [Sdoi2011]消耗战 虚树 树形dp

2286: [Sdoi2011]消耗战

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 4469  Solved: 1646
[Submit][Status][Discuss]

Description

在一场战争中，战场由n个岛屿和n-1个桥梁组成，保证每两个岛屿间有且仅有一条路径可达。现在，我军已经侦查到敌军的总部在编号为1的岛屿，而且他们已经没有足够多的能源维系战斗，我军胜利在望。已知在其他k个岛屿上有丰富能源，为了防止敌军获取能源，我军的任务是炸毁一些桥梁，使得敌军不能到达任何能源丰富的岛屿。由于不同桥梁的材质和结构不同，所以炸毁不同的桥梁有不同的代价，我军希望在满足目标的同时使得总代价最小。

侦查部门还发现，敌军有一台神秘机器。即使我军切断所有能源之后，他们也可以用那台机器。机器产生的效果不仅仅会修复所有我军炸毁的桥梁，而且会重新随机资源分布（但可以保证的是，资源不会分布到1号岛屿上）。不过侦查部门还发现了这台机器只能够使用m次，所以我们只需要把每次任务完成即可。

Input

第一行一个整数n，代表岛屿数量。

接下来n-1行，每行三个整数u,v,w，代表u号岛屿和v号岛屿由一条代价为c的桥梁直接相连，保证1<=u,v<=n且1<=c<=100000。

第n+1行，一个整数m，代表敌方机器能使用的次数。

接下来m行，每行一个整数k_i，代表第i次后，有k_i个岛屿资源丰富，接下来k个整数h₁,h₂,…h_k，表示资源丰富岛屿的编号。

Output

输出有m行，分别代表每次任务的最小代价。

 

Sample Input

10
1 5 13
1 9 6
2 1 19
2 4 8
2 3 91
5 6 8
7 5 4
7 8 31
10 7 9
3
2 10 6
4 5 7 8 3
3 9 4 6

Sample Output

12
32
22

HINT

 对于100%的数据，2<=n<=250000,m>=1,sigma(ki)<=500000,1<=ki<=n-1

这是一道虚树裸题

关于虚树的讲解可以看这里

虚树的概念

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<map>
#include<set>
using namespace std;

typedef long long ll;

inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=10*x+ch-'0';ch=getchar();}
	return x*f;
}
void print(ll x)
{if(x<0)putchar('-'),x=-x;if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=250100,inf=0X3f3f3f3f;

int last[N],ecnt;
struct EDGE{int to,nt,val;}e[N<<1];
inline void add(int u,int v,int val)
{if(u==v) return ;e[++ecnt]=(EDGE){v,last[u],val};last[u]=ecnt;}

int n;

ll mn[N];
int dfn[N],tim;
int dep[N],anc[N][20];

void init(int u)
{
	dfn[u]=++tim;
	for(int i=1;(1<<i)<=dep[u];++i)
		anc[u][i]=anc[anc[u][i-1]][i-1];
	for(int i=last[u],v;i;i=e[i].nt)
		if((v=e[i].to)!=anc[u][0])
		{
			mn[v]=min(mn[u],1ll*e[i].val);
			dep[v]=dep[u]+1;
			anc[v][0]=u;
			init(v);
		}
}

inline int getlca(int u,int v)
{
	if(dep[u]<dep[v]) swap(u,v);
	int len=dep[u]-dep[v];
	for(int i=0;(1<<i)<=len;++i)
		if((1<<i)&len)
			u=anc[u][i];
	if(u==v) return u;
	for(int i=18;~i;--i)
		if(anc[u][i]!=anc[v][i])
			u=anc[u][i],v=anc[v][i];
	return anc[u][0];
}

ll f[N];

void dp(int u)
{
	f[u]=0;
	for(int i=last[u],v;i;i=e[i].nt)
	{
		v=e[i].to;dp(v);
		f[u]+=f[v];
	}
	last[u]=0;
	!f[u] ? f[u]=mn[u] : f[u]=min(f[u],mn[u]);	
}

int h[N];

inline bool cmp(int x,int y)
{return dfn[x]<dfn[y];}

int st[N];

void solve()
{
	ecnt=0;
	int K=read();
	for(int i=1;i<=K;++i) h[i]=read();
	sort(h+1,h+1+K,cmp);
	int tot(1),top(1);
	for(int i=2;i<=K;++i)
		if(getlca(h[tot],h[i])!=h[tot])
			h[++tot]=h[i];
	st[top]=1;
	for(int i=1,grd;i<=tot;++i)
	{
		grd=getlca(h[i],st[top]);
		while(1)
		{
			if(dep[st[top-1]]<=dep[grd])
			{
				add(grd,st[top],0);top--;
				if(st[top]!=grd) st[++top]=grd;
				break;
			}
			add(st[top-1],st[top],0);top--;
		}
		if(st[top]!=h[i]) st[++top]=h[i];
	}
	while(--top) add(st[top],st[top+1],0);
	dp(1);
	print(f[1]);putchar('\n');
}

int main()
{
	n=read();
	register int i,u,v,val;
	for(i=1;i<n;++i)
		u=read(),v=read(),val=read(),
		add(u,v,val),add(v,u,val);
	mn[1]=1ll*inf*inf;init(1);
	memset(last,0,sizeof(last));
	register int Q=read();
	while(Q--) solve();
	return 0;
}