Codeforces Round 928 (Div. 4)G. Vlad and Trouble at MIT 树上dp

Problem - G - Codeforces

题意：

给你一棵树，书上节点有三种状态：S 睡觉， P 开party ，C 随意

开party的声音会传播到所有相邻节点（包括相邻的相邻节点），所以本题要求你在一些通路上做墙，问最少墙数。

思路：

首先这是棵树，我作为新手，当图来做，当然非常复杂啊，“贪心不能解决的情况”。很不系统。

所以树的话要上下做。

问最大最小可以想dp。

（还有就是本题给树的方式特殊，是第i个节点的父节点这样，也是见这位up的邻接表不双向才发现Codeforces Round 928讲解(BCDEG)_哔哩哔哩_bilibili）

————

树上dp，我们先不考虑C节点，也就是只有S 和 P。

这样状态方程很简单。（这个都想不到，我就太捞了）

dfs(int cur)
{
    dp[cur] = 0;   
    for(int x : alist[cur])
    {
        dfs(x)
        if(cur和x子节点的状态相同)
            dp[cur] += dp[x];
        else//和子节点状态不同，需要建一堵墙
            dp[cur] += dp[x] + 1;
    }
}

————

然后就是加上C，因为C既可以接受party的声音，也可以安静，不过会传播party的声音。

所以我们可以把C看做P或者S，都可以，所以取墙数少的

我们可以开二维dp来囊括这所有情况，很容易看懂的，直接上代码了：

（判断和子节点状态是否相同可以“先搁置”，就用0和1代表现在状态，最后再将非法状态设为大数即可）

vector<vector<int>>alist;
char sta[100005]; //状态 C S P
int dp[100005][2];
//0 s
//1 party
void dfs(int cur)
{
	dp[cur][0] = dp[cur][1] = 0;
	for (auto x : alist[cur])
	{
		dfs(x);
		if (sta[x] == 'C')
		{
			dp[cur][0] += min(dp[x][0],dp[x][1]+1);
			dp[cur][1] += min(dp[x][0]+1,dp[x][1]);
		}
		else if (sta[x] == 'S')
		{
			dp[cur][0] += dp[x][0];
			dp[cur][1] += dp[x][0] + 1;
		}
		else if (sta[x] == 'P')
		{
			dp[cur][0] += dp[x][1] + 1;
			dp[cur][1] += dp[x][1];
		}
	}

    //非法情况设为最大值，不取
	if (sta[cur] == 'S')dp[cur][1] = 1e9;
	if (sta[cur] == 'P')dp[cur][0] = 1e9;
}

参考代码：

vector<vector<int>>alist;
char sta[100005];
int dp[100005][2];
//0 s
//1 party
void dfs(int cur)
{
	dp[cur][0] = dp[cur][1] = 0;
	for (auto x : alist[cur])
	{
		dfs(x);
		if (sta[x] == 'C')
		{
			dp[cur][0] += min(dp[x][0],dp[x][1]+1);
			dp[cur][1] += min(dp[x][0]+1,dp[x][1]);
		}
		else if (sta[x] == 'S')
		{
			dp[cur][0] += dp[x][0];
			dp[cur][1] += dp[x][0] + 1;
		}
		else if (sta[x] == 'P')
		{
			dp[cur][0] += dp[x][1] + 1;
			dp[cur][1] += dp[x][1];
		}
	}
	if (sta[cur] == 'S')dp[cur][1] = 1e9;
	if (sta[cur] == 'P')dp[cur][0] = 1e9;
}
void solve()
{
	int n;
	cin >> n;
	alist = vector<vector<int>>(n + 1);
	for (int i = 2; i <= n; i++)
	{
		int tmp;
		cin >> tmp;
		alist[tmp].push_back(i);
		//alist[tmp].push_back(i); 这种按顺序单给边的情况
		//我接到谁上，我是子树。 （必须有人接1，也必须有人接我？但确实是每个边都必有
	}
	for (int i = 1; i <= n; i++)
	{
		cin >> sta[i];
	}
	dfs(1);
	cout << min(dp[1][0], dp[1][1]) << endl;
}
signed main()
{
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	int t = 1;
	cin >> t;
	while (t--)
	{
		solve();
	}
	return 0;
}