1. 程式人生 > >【bzoj1369】[Baltic2003]Gem(樹形dp+結論)

【bzoj1369】[Baltic2003]Gem(樹形dp+結論)

  題目傳送門:bzoj1369

  這題其實有個結論:節點數為n的樹,對其染色使相鄰節點顏色不同,且總顏色權值最小,所需的顏色數量是$ O(\log n) $的。

  所以我們就可以愉快的dp了:$ f[i][j] $表示處理到以$ i $為根的子樹,根節點顏色為$ j $的最小總價值,然後就可以隨便轉移了。

  程式碼:

#include<cstdio>
#include<cstring>
#define maxn 10010
struct edge{
    int to,nxt;
}e[2*maxn];
int fir[maxn],f[maxn][20
]; int n,tot=0; void add(int x,int y){e[tot].to=y; e[tot].nxt=fir[x]; fir[x]=tot++;} void dfs(int now,int fa) { for(int i=fir[now];~i;i=e[i].nxt) if(e[i].to!=fa)dfs(e[i].to,now); for(int i=1;i<=10;i++){ f[now][i]=i; for(int j=fir[now];~j;j=e[j].nxt)
if(e[j].to!=fa){ int mn=0x3f3f3f3f; for(int k=1;k<=10;k++) if(k!=i&&f[e[j].to][k]<mn)mn=f[e[j].to][k]; f[now][i]+=mn; } } } int main() { memset(fir,255,sizeof(fir)); tot=0; scanf("%d",&n);
for(int i=1;i<n;i++){ int x,y; scanf("%d%d",&x,&y); add(x,y); add(y,x); } dfs(1,-1); int ans=0x3f3f3f3f; for(int i=1;i<=10;i++) if(f[1][i]<ans)ans=f[1][i]; printf("%d\n",ans); }
bzoj1369