傳送門(mén)

我一開(kāi)始想的是考慮每個(gè)點(diǎn)的顏色
設(shè)的狀態(tài)就是\(f[i][j]\)表示\(i\)子樹(shù)里有\(j\)個(gè)黑點(diǎn)的\(i\)子樹(shù)的收益最大值，后來(lái)發(fā)現(xiàn)無(wú)法轉(zhuǎn)移
那么考慮答案的統(tǒng)計(jì)，可以對(duì)于邊統(tǒng)計(jì)答案
那么我們就可以考慮\(f[i][j]\)為\(i\)子樹(shù)里有\(j\)個(gè)黑點(diǎn)對(duì)于全局答案的貢獻(xiàn)最大值
也就是對(duì)于邊考慮統(tǒng)計(jì)答案，假設(shè)邊的長(zhǎng)度是\(val\)
那么轉(zhuǎn)移方程就是：
\(f[x][i]=max\{f[x][i],f[son][j]+f[x][i-j]+val*j*(k-j)+val*(size[son]-j)*(n-size[son]-k+j)\}\)
我代碼挺慢的，卡過(guò)去的！
代碼：

#include<cstdio> #include<iostream> #include<algorithm> #include<cstring> using namespace std; void read(int &x) {char ch;bool ok;for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar());if(ok) x=-x; } #define rg register const int maxn=2e3+10; int n,m,pre[maxn*2],nxt[maxn*2],h[maxn],v[maxn*2],cnt,size[maxn]; long long f[maxn][maxn]; void add(int x,int y,int z) {pre[++cnt]=y,nxt[cnt]=h[x],h[x]=cnt,v[cnt]=z;pre[++cnt]=x,nxt[cnt]=h[y],h[y]=cnt,v[cnt]=z; } void dfs(int x,int fa) {size[x]=1;f[x][0]=f[x][1]=0;for(rg int i=h[x];i;i=nxt[i])if(pre[i]!=fa){dfs(pre[i],x);size[x]+=size[pre[i]];}for(rg int i=h[x];i;i=nxt[i])if(pre[i]!=fa)for(rg int j=min(m,size[x]);j>=0;j--)for(rg int k=0;k<=min(j,size[pre[i]]);k++)if(~f[x][j-k])f[x][j]=max(f[x][j],f[pre[i]][k]+f[x][j-k]+1ll*k*(m-k)*v[i]+1ll*(size[pre[i]]-k)*(n-size[pre[i]]-m+k)*v[i]); } signed main() {read(n),read(m);memset(f,-1,sizeof f);for(rg int i=1,x,y,z;i<n;i++)read(x),read(y),read(z),add(x,y,z);dfs(1,0),printf("%lld\n",f[1][m]); }

轉(zhuǎn)載于:https://www.cnblogs.com/lcxer/p/10495178.html

總結(jié)

以上是生活随笔為你收集整理的bzoj4033:[HAOI2015]树上染色的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。