正題

題目鏈接:https://www.luogu.com.cn/problem/P3177

---

題目大意

$n$個點的一棵樹，將$k$個點染成黑色，其他是白色，使得黑點之間和白點之間兩兩距離和最大。

---

解題思路

也是統計每條邊的貢獻，設$f_{x,i}$表示節點$x$的子樹中有$i$個黑點時的最大子樹貢獻和。

發現這樣轉移是$O(n{k}^2)$的，只要限制$i$的枚舉范圍在$si{z}_x$以內就變成$O(n^2)$了，原因是節點之間兩兩做了一次貢獻。

---

$code$

#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const ll N=2100; struct node{ll to,next,w; }a[N*2]; ll n,K,tot,ls[N],f[N][N],g[N],siz[N]; void addl(ll x,ll y,ll w){a[++tot].to=y;a[tot].next=ls[x];ls[x]=tot;a[tot].w=w;return; } void dp(ll x,ll fa){siz[x]=1;for(ll i=ls[x];i;i=a[i].next){ll y=a[i].to;if(y==fa)continue;dp(y,x);for(ll j=0;j<=siz[x];j++)g[j]=f[x][j],f[x][j]=0;for(ll j=0;j<=siz[x];j++)for(ll k=0;k<=siz[y];k++)f[x][j+k]=max(f[x][j+k],g[j]+f[y][k]+a[i].w*(k*(K-k)+(siz[y]-k)*(n-K-siz[y]+k))); siz[x]+=siz[y];}return; } int main() {scanf("%lld%lld",&n,&K);for(ll i=1;i<n;i++){ll x,y,w;scanf("%lld%lld%lld",&x,&y,&w);addl(x,y,w);addl(y,x,w);}dp(1,1);printf("%lld",f[1][K]); }

總結

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

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