1099 -- 【CQOI2005】珠宝
Description
给一棵n个结点的树,给每个点安排一个正整数编号,使得相邻点具有不同的编号,编号的总和尽量小。
Input
第一行:n(n<=50,000)
以下n-1行,每行两个数u,v(1<=u,v<=n),表示u 和v有一条边
Output
仅一行,为最小编号和
Sample Input
8
1 2
1 3
1 4
1 5
5 6
5 7
5 8
Sample Output
11
一道裸的树形dp,但是我们应该怎么去考虑他的编号呢?
我们可以预见到,这些正整数编号应该都不会太大,所以直接枚举我的编号是多少和我儿子的编号是多少就可以了,详见代码。
枚举的编号应该还可以再小一点,可以去试一试,应该在9都可以。
枚举的编号应该还可以再小一点,可以去试一试,应该在9都可以。
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
using namespace std;
struct node{int to,nxt,val;
}w[200005];
int cnt=0,f[100005][30],h[100005],n;
void add(int x,int y)
{
cnt++;w[cnt].to=y;w[cnt].nxt=h[x];h[x]=cnt;
}
void dfs(int pos,int fa)
{
if(f[pos][1])return ;
for(int k=1;k<=30;k++)
{
for(int i=h[pos];i;i=w[i].nxt)
{
int to=w[i].to;
if(to==fa)continue;
dfs(to,pos);
int minn=0x3fffffff;
for(int j=1;j<=30;j++)
if(j!=k&&f[to][j]<minn)minn=f[to][j];
f[pos][k]+=minn;
}
f[pos][k]+=k;
}
}
int main(){
cin>>n;
int a,b,c;
for(int i=1;i<=n-1;i++)
{
cin>>a>>b;
add(a,b);
add(b,a);
}
dfs(1,0);
int ans=0x3fffffff;
for(int i=1;i<=30;i++)
ans=min(ans,f[1][i]);
cout<<ans;
return 0;
}