ABC 368D Minimum Steiner Tree

題意
給你一顆由N個點組成的樹，指定K個節點，求包含這K個節點的最小子樹的大小

思路
考慮正難則反，我們從開始的樹當中剪掉那些沒有任何指定點的子樹，剩下來的子樹就是最小的、能包含所有指定節點的子樹。關於剪去這個操作，就是dfs一旦遇到以當前節點為根的子樹沒有任何指定點時，就停止dfs，並把該子樹的大小貢獻到much裡。最後用n減去much，就是答案（注意選根的時候要注意，用指定的某個點作為根來進行dfs，否則這個方法會被菊花圖給hack掉）

程式碼

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=2e5+10;
int a[maxn],sz[maxn],much;
vector<int>ds[maxn];

void dfs1(int u,int fa)
{
	sz[u]=1;
	for(int i=0;i<ds[u].size();i++)
	{
		int v=ds[u][i];
		if(v==fa) continue;
		dfs1(v,u);
		sz[u]+=sz[v];
		a[u]+=a[v];
	}
}

void dfs2(int u,int fa)
{
	if(!a[u])
	{
		much+=sz[u];
		return ;
	}
	for(int i=0;i<ds[u].size();i++)
	{
		int v=ds[u][i];
		if(v==fa) continue;
		dfs2(v,u);
	}
}


signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	
	int n,k;
	cin>>n>>k;
	for(int i=1;i<=n-1;i++)
	{
		int u,v;
		cin>>u>>v;
		ds[u].push_back(v);
		ds[v].push_back(u);
	}
	int begin;
	for(int i=1;i<=k;i++) 
	{
		int x;
		cin>>x;
		begin=x;
		a[x]=1;
	}
	dfs1(begin,0);
	dfs2(begin,0);
	cout<<n-much<<endl;
	
	
	return 0;
}