1369: [Baltic2003]Gem

题目链接

题目大意：给为树上的结点标上权值，权值可以是任意的正整数，相邻的两个结点不能标上相同的权值，求最小权值和

题解：f[i][j]表示以i为根的子树i的权值为j的最小总权值，容易YY出颜色大概是log级别的，然后乱搞，枚举x的颜色，x的儿子取和x不同色的最小值……

我的收获：结论，结论……

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define B 18
#define INF 0x3f3f3f3f
const int M=10005;

int n,t,ans=INF;
int f[M][B],head[M];

struct edge{int to,nex;}e[M*4];

void add(int u,int v){e[t].to=v,e[t].nex=head[u],head[u]=t++;}

void dfs(int x,int fa)
{
    for(int i=1;i<=B;i++) f[x][i]=i;
    for(int i=head[x];i!=-1;i=e[i].nex){
        int v=e[i].to;
        if(v==fa) continue;
        dfs(v,x);
        for(int j=1;j<=B;j++){
            int mi=INF;
            for(int k=1;k<=B;k++)
            if(k!=j) mi=min(mi,f[v][k]);
            f[x][j]+=mi;
        }
    }
}

void work()
{
    dfs(1,0);
    for(int i=1;i<=B;i++) ans=min(ans,f[1][i]);
    cout<<ans<<endl;
}
void init()
{
    int x,y;
    t=0;memset(head,-1,sizeof(head));
    cin>>n;
    for(int i=1;i<n;i++) 
        scanf("%d%d",&x,&y),add(x,y),add(y,x);
}

int main()
{
    init();
    work();
}