【bzoj2286】[Sdoi2011消耗战 虚树+dp

学一下虚树

单次dp是O(N)的，总复杂度O(NM)

但实际用不到O(N)个节点，只有O(K)个节点是有用的，若能建出一棵O(K)个节点的树，那么复杂度会变成O(∑K)

把给定节点按照dfs序排序，维护一个栈，栈中的节点是根到栈顶元素的链上的节点

加入节点x，设栈顶元素为p，t=LCA(p,x)

若t==p，则直接加入x

若t!=p，则找到栈中最后一个深度小于t的节点y，在y到p之间的节点中，不会再有中间节点插入了，将这些点删掉，对应从st[top-1]向st[top]连边，最后将点x加入

比较有趣的东西

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<vector>
#define maxn 250010

using namespace std;

int head[maxn],to[2*maxn],next[2*maxn],len[2*maxn];
long long dp[maxn],val[maxn];
vector<int> v[maxn];
bool flag[maxn];
int in[maxn],st[maxn],dep[maxn],fa[20][maxn],a[maxn],q[maxn];
int n,m,tot,num,top,T,k,now;

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

void addedge(int x,int y,int z)
{
	num++;to[num]=y;len[num]=z;next[num]=head[x];head[x]=num;
}

void dfs(int x)
{
	in[x]=++tot;
	for (int p=head[x];p;p=next[p])
	  if (to[p]!=fa[0][x]) fa[0][to[p]]=x,dep[to[p]]=dep[x]+1,val[to[p]]=min(val[x],(long long)len[p]),dfs(to[p]);
}

void DP(int x)
{
	if (flag[x]) dp[x]=val[x];
	else
	{
		dp[x]=0;
		for (int i=0;i<v[x].size();i++) DP(v[x][i]),dp[x]+=dp[v[x][i]];
		dp[x]=min(dp[x],val[x]);
	}
}

void DFS(int x)
{
	for (int i=0;i<v[x].size();i++) DFS(v[x][i]);
	v[x].clear();flag[x]=0;
}

int go_up(int x,int d)
{
	for (int i=18;i>=0;i--)
	  if (d&(1<<i)) x=fa[i][x];
	return x;
}

int LCA(int x,int y)
{
	if (dep[x]>dep[y]) x=go_up(x,dep[x]-dep[y]);
	else y=go_up(y,dep[y]-dep[x]);
	if (x==y) return x;
	for (int i=18;i>=0;i--)
	  if (fa[i][x]!=fa[i][y]) x=fa[i][x],y=fa[i][y];
	return fa[0][x];
}

int main()
{
	scanf("%d",&n);
	for (int i=1;i<n;i++)
	{
		int x,y,z;
		scanf("%d%d%d",&x,&y,&z);
		addedge(x,y,z);addedge(y,x,z);
	}
	val[1]=1e18;dfs(1);
	for (int j=1;j<=18;j++)
	  for (int i=1;i<=n;i++)
	    fa[j][i]=fa[j-1][fa[j-1][i]];
	scanf("%d",&T);
	while (T--)
	{
		int x;
		top=1;st[top]=1;
		scanf("%d",&k);
		for (int i=1;i<=k;i++) scanf("%d",&a[i]),flag[a[i]]=1;
		sort(a+1,a+k+1,cmp);
		for (int i=1;i<=k;i++)
		{
			int lca=LCA(a[i],st[top]);
			while (dep[st[top]]>dep[lca])
				if (dep[st[top-1]]<dep[lca]) v[lca].push_back(st[top]),st[top]=lca;
				else v[st[top-1]].push_back(st[top]),top--;
			if (st[top]!=a[i]) st[++top]=a[i];
		}
		while (top>1) v[st[top-1]].push_back(st[top]),top--;
		DP(1);
		DFS(1);
		printf("%lld\n",dp[1]);
	}
	return 0;
}