CF613D Kingdom and its Cities

重点在树形dp。

dp[u]:u为根的子树的答案最小值

对于有些只有一个重要节点的子树，可以暂时保留这个节点 ，等到不得已必须处理时再处理，可以使得答案更优

所以再记录一个g[u]:u为根的子树，当答案最小时，还剩下没处理好的节点个数

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,u,v,q,k;
int tot,dfn[N],low[N],size[N],d[N],f[N],son[N],top[N];
bool vis[N];
int p[N<<1],sta[N],dp[N],g[N];
int cnt,head[N];
struct edge{int next,to;}e[N<<1]; 

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

void dfs(int u,int fa)
{
	size[u]=1;
	dfn[u]=++tot;
	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;
		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;
}

void DP(int u)
{
	for (register int i=head[u]; i; i=e[i].next)
	{
		DP(e[i].to);
		dp[u]+=dp[e[i].to]; g[u]+=g[e[i].to];
	}
	if (vis[u]) dp[u]+=g[u],g[u]=1;
	else dp[u]+=(g[u]>1),g[u]=(g[u]==1);
}

inline int solve(int u)
{
	for (register int i=1; i<=k; ++i) if (vis[p[i]] && vis[f[p[i]]]) return -1;
	DP(u);
	return dp[u];
}

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

int main(){
	scanf("%d",&n);
	for (register int i=1; i<n; ++i) scanf("%d%d",&u,&v),add(u,v),add(v,u);
	dfs(1,0); dfs2(1,1);
	scanf("%d",&q);
	memset(head,0,sizeof(head));
	while (q--)
	{
		scanf("%d",&k);
		for (register int i=1; i<=k; ++i) scanf("%d",&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;
		cnt=0;
		for (register int i=1,top=0; i<=k; ++i)
		{
			while (top && low[sta[top]]<dfn[p[i]]) top--;
			add(sta[top],p[i]); sta[++top]=p[i];
		}
		printf("%d\n",solve(p[1]));
		for (register int i=1; i<=k; ++i) head[p[i]]=0,vis[p[i]]=false,dp[p[i]]=g[p[i]]=0;
	}
return 0;
}