牛客月赛3-旅游-(树形dp)

G

题意：
就是给你n个城市，这个城市由n-1条边连着，然后初始点在m，在m点住宿，然后观光所有距离和他为1的城市，然后第二天再选择一个城市去居住，但是不能是观光过的。问你最多可以玩多少天。

思考：
刚开始看到，我以为是图的问题，但是感觉太不像了，没算法能做，我一看n-1那很明显就是树了，然后呢发现，距离和他为1的点不就是他的儿子吗，所以经典的树形dp，就和以前做过的没有上司的舞会一样。所以dp[i][j]代表到i这个节点，1代表选这个点，0代表不选，所以dp一下就行了。值得注意的是，初始点是必选的，所以最后答案就是dp[m][1]，而不是和dp[m][0]取max。

代码：

int T,n,m,k;
int va[N];
int dp[N][2];

vector<int > e[N];

void dfs(int now,int p)
{
	dp[now][1] = 1;
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		dfs(spot,now);
		dp[now][1] = max(dp[now][1],dp[now][1]+dp[spot][0]);
		dp[now][0] = max(dp[now][0],dp[now][0]+max(dp[spot][0],dp[spot][1]));
	}
}

signed main()
{
	IOS;
	cin>>n>>m;
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		e[a].pb(b);
		e[b].pb(a);
	}
	dfs(m,0);
	cout<<dp[m][1];
	return 0;
}

总结：
多多思考呀，和以前的算法练习起来。