Machine Learning on a Tree ZOJ - 4007

我的ma，前面的题意这么吓人

其实比区间dp简单多了 

给你一颗树然后，树上的点有权值1,0,-1，-1可以换成0或者1，然后每条变连接的两个点权值如果不一样那么答案加一，让你求把1怎么变使得最后的答案最小

然后么dfs搜到最底层 然后开始从底层递推到树顶

 

怎么递推？

先预处理

if(a[u]!=-1) dp[u][a[u]]=0,dp[u][a[u]^1]=inf;
	else dp[u][0]=dp[u][1]=0;

对于-1节点是可以修改的 那么它在本层u就不会增加贡献 所以他的贡献就设为0；

对于1或者0节点无法修改 那么自身的贡献设为0 到改为另一个的贡献设为inf 那么在min的时候就不会选到另外一条路径；

然后由dfs的特性自底向上递推

dp[u][1]+=min(dp[to][1],dp[to][0]+1);
		dp[u][0]+=min(dp[to][0],dp[to][1]+1);

在本层u选上下一层贡献最少的 （为嘛这样做？ 因为我们是自底向上递推的啊，底层情况都是已经推出来的，其中最底层的情况是预处理好的）

if(fa==to) continue;

这个是边界 

 

 

然后再main中

if(a[1]==-1) printf("%d\n",min(dp[1][0],dp[1][1]));

这个的意思是当我的1节点是可更改的（-1） 那么就找最小的

反之 不可更改那就直接输出1节点的值咯

 

讲这么清楚了 赶紧去a题吧 XD

#include<cstdio>
#include<vector>
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=1e5+10;
vector<int>v[maxn];
int dp[maxn][2];
int a[maxn];

void dfs(int u,int fa)
{
	if(a[u]!=-1) dp[u][a[u]]=0,dp[u][a[u]^1]=inf;
	else dp[u][0]=dp[u][1]=0;
	for(int i=0;i<v[u].size();i++)
	{
		int to=v[u][i];
		if(fa==to) continue;
		dfs(to,u);
		dp[u][1]+=min(dp[to][1],dp[to][0]+1);
		dp[u][0]+=min(dp[to][0],dp[to][1]+1);
	}
}

int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		int n;
		scanf("%d",&n);
		for(int i=1;i<=n;i++) scanf("%d",&a[i]),v[i].clear();
		for(int i=0;i<n-1;i++)
		{
			int s,e;
			scanf("%d%d",&s,&e);
			v[s].push_back(e);
			v[e].push_back(s);
		}
		dfs(1,-1);
		if(a[1]==-1) printf("%d\n",min(dp[1][0],dp[1][1]));
		else printf("%d\n",dp[1][a[1]]);
	}
}