简化题意,给一棵树,找出恰好 \(k+1\) 条链,是这些链之和最大。
有恰好选出的字眼,并且原问题显然具有凸性,直接考虑 wqs 二分。
然后每条链会减去二分的 \(mid\),然后没有限制,求最大链和及链的数量,考虑树形 dp。
设 \(f_{x,0/1/2}\) 表示以 \(x\) 为根的子树,\(x\) 点入度为 \(0/1/2\) 所得的最大链值。
特别地,每次转移完 \(x\) 后令 \(f_{x,0}=\max(f_{x,0},f_{x,1}+mid,f_{x,2})\),表示 \(x\) 不再和祖先进行链的合并,此时得到的最优解。
接下来就能转移了,设 \(y\) 是 \(x\) 的儿子,转移分三步:
\(f_{x,2}=\max(f_{x,2}+f_{y,0},f_{x,1}+f_{y,1}+w_{(x,y)}+mid)\)
\(f_{x,1}=\max(f_{x,1}+f_{y,0},f_{x,0}+f_{y,1}+w_{(x,y)})\)
\(f_{x,0}=\max(f_{x,0}+f_{y,0})\)
最后判断链数和 \(k\) 的大小关系,就做完了。
注意边权相等时,链数少的状态优先转移,因为维护的是上凸包。
#include<bits/stdc++.h>
using namespace std;
#define rd read()
#define ll long long
#define FOR(i,j,k) for(int i=j;i<=k;i++)
#define ROF(i,j,k) for(int i=j;i>=k;i--)
int read(){
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch)) x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
return x*f;
}
const int N=3e5+10;
const ll INF=1e12;
int n,k,head[N],tot;
struct node{int to,nxt,w;}edge[N<<1];
void add(int x,int y,int w){edge[++tot]={y,head[x],w};head[x]=tot;}
struct Node{
ll val;int cnt;
Node(ll x=0,int y=0):val(x),cnt(y){}
friend bool operator<(Node a,Node b){return a.val!=b.val?a.val<b.val:a.cnt>b.cnt;}
friend Node operator+(Node a,Node b){return Node{a.val+b.val,a.cnt+b.cnt};}
friend Node operator+(Node a,ll v){return Node{a.val+v,a.cnt};}
}f[N][3],tmp;
void dfs(int x,int fa){
f[x][0]=f[x][1]=f[x][2]=Node();
f[x][2]=max(f[x][2],tmp);
for(int i=head[x];i;i=edge[i].nxt){
int y=edge[i].to;if(y==fa) continue;
dfs(y,x);
f[x][2]=max(f[x][2],max(f[x][2]+f[y][0],f[x][1]+f[y][1]+edge[i].w+tmp));
f[x][1]=max(f[x][1],max(f[x][1]+f[y][0],f[x][0]+f[y][1]+edge[i].w));
f[x][0]=max(f[x][0],f[x][0]+f[y][0]);
}
f[x][0]=max(f[x][0],max(f[x][1]+tmp,f[x][2]));
}
int main(){
n=rd,k=rd,k++;
FOR(i,1,n-1){int x=rd,y=rd,w=rd;add(x,y,w),add(y,x,w);}
ll l=-INF,r=INF,ans=0 ;
while(l<r){
ll mid=(l+r+1)/2ll;
tmp=Node{-mid,1};dfs(1,0);
if(f[1][0].cnt==k){printf("%lld\n",f[1][0].val+k*mid);return 0;}
if(f[1][0].cnt>k) l=mid+1;
else r=mid;
}
tmp=Node{-l,1};dfs(1,0),printf("%lld\n",f[1][0].val+k*l);
return 0;
}
标签:Node,八省,ch,int,max,mid,rd,2018,联考
From: https://www.cnblogs.com/summ1t/p/18527828