BZOJ1369: [Baltic2003]Gem 树形DP

最新推荐文章于 2022-05-18 07:57:43 发布

Oakley_

最新推荐文章于 2022-05-18 07:57:43 发布

阅读量513

点赞数 1

文章标签： BZOJ 树形DP

本文链接：https://blog.csdn.net/Oakley_/article/details/52687542

版权

BZOJ 1369: [Baltic2003]Gem

Time Limit: 2 Sec Memory Limit: 64 MB

Submit: 343 Solved: 219

[Submit][Status][Discuss]

题解：
TreeDP，每个点的颜色大概8种就够了

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=10005;
const int M=20005;
int cnt,to[M],lj[N],nxt[M];
void insert(int f,int t)
{
	cnt++,to[cnt]=t,nxt[cnt]=lj[f],lj[f]=cnt;
	cnt++,to[cnt]=f,nxt[cnt]=lj[t],lj[t]=cnt;
}
int n,f[N][9],fa[N];
void dfs(int x)
{
	for(int i=lj[x];i;i=nxt[i])
	if(to[i]!=fa[x])
	{
		fa[to[i]]=x;
		dfs(to[i]);
		for(int j=1;j<=8;j++)
		{
			int mn=1e9;
			for(int k=1;k<=8;k++)
			if(k!=j) mn=min(mn,f[to[i]][k]);
			f[x][j]+=mn;
		}
	}
	for(int i=1;i<=8;i++) f[x][i]+=i;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		insert(x,y);
	}
	dfs(1);
	int ans=1e9;
	for(int i=1;i<=8;i++) ans=min(ans,f[1][i]);
	printf("%d",ans);
}