傳送門

我一開始想的是考慮每個點的顏色
設的狀態就是\(f[i][j]\)表示\(i\)子樹里有\(j\)個黑點的\(i\)子樹的收益最大值，后來發現無法轉移
那么考慮答案的統計，可以對于邊統計答案
那么我們就可以考慮\(f[i][j]\)為\(i\)子樹里有\(j\)個黑點對于全局答案的貢獻最大值
也就是對于邊考慮統計答案，假設邊的長度是\(val\)
那么轉移方程就是：
\(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)\}\)
我代碼挺慢的，卡過去的！
代碼：

#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]); }

轉載于:https://www.cnblogs.com/lcxer/p/10495178.html

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。