2019.02.26【九省联考2018】【BZOJ5252】【洛谷P4383】林克卡特树（DP凸优化）

BZOJ传送门

洛谷传送门

---

解析：

首先让我们好好分析一下题目。

连上$k$条边权为$0$的边。我们发现这个操作并不会产生任何贡献？？？

于是问题就是最大化树上不超过$k+1$条不相交的链的边权和。

如果定义单点是退化的链，那就是恰好$k+1$条。

接下来的讨论都是建立在这个定义上的。

这个问题相信敢来写九省联考题的人都会用树形DP解决。

于是你可以拿到$60pts$的暴力了。

然后官方题解（据说）里面有这么一句话：假如你是个秒出60pts的巨佬，即将AK之时闲来无事输出选了恰好k条链的最优解，你就会发现：最优解数组是一个上凸函数。

。。。

WTM。。。什么玩意。。。

换句话说，这个玩意天生的就是一个凸壳，不过我们不知道它长什么样子。

为什么是一个凸壳？可以感性理解一下。

首先一定存在$k_0$使得选择恰好$k_0$条链能够得到最优解，然后证明，对于$k_0$两边的凸壳，结果一定单调。

定义$f(k)$为恰好选择$k$条边的时候的最优解。

考虑证明$k<k_0$的情况，另一边同理 应该 也证明得出来，我没试过。

如果对于$k_1<k_2<k_0$，$f(k_1)>f(k_2)$，说明我们用更少的链凑出了更大的权值和。

首先明确一点，最优解$k_0$的方案中一定没有负边，不然我们总是可以通过增加$k_0$而删除负边来使得总贡献更大。

同时由于$k_0$是最优解，所有的正边它都是包含了的。

那么如果在$k_1<k_2<k_0$的时候$f(k_1)>f(k_2)$，说明凭空多出了在较优解中的正边，那么$f(k_2)$就不是$k_2$时候的最优解。

于是我们考虑找到这个凸包上的$(k,f(k))$一点。

所以现在我们需要让这个点变得“突出”。

换句话说，我们要通过某种简单变换，使得这个点在凸壳上最高。

想到了什么，二分斜率！

我们反过来，二分斜率后，设$g(k)=f(k)-mid\ast k$，就是每条链的权值减去一个$mid$。

如果我们二分到了恰当的斜率，那么这时候的最优解$(k,g(k))$就是$g$形成的凸壳的最高点。

我们只需要看$g$的最高点的横坐标在$k$的左还是右就可以调整斜率了。

我们利用接下来叙述的DP方式，可以将这个最高点的横纵坐标求出来，还原到$f$。

考虑每一个点在最后的方案中只有三种状态：链端，链中，孤立。

特别的，退化的孤立点视为链中，即它不能与其他位置结合形成新的链。

然后这些情况可以用度数来轻易表示出。

于是愉快的DP出选择的最优解和选出最优解的时候最小的链的数量就行了。

---

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc get_char
#define cs const

namespace IO{
	inline char get_char(){
		static cs int Rlen=1<<20|1;
		static char buf[Rlen],*p1,*p2;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
	}
	
	inline int getint(){
		re char c;
		re bool f=0;
		while(!isdigit(c=gc()))if(c=='-')f=1;re int num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return f?-num:num;
	}
}
using namespace IO;

cs int N=300005;

int last[N],nxt[N<<1],to[N<<1],w[N<<1],ecnt;
inline void addedge(int u,int v,int val){
	nxt[++ecnt]=last[u],last[u]=ecnt,to[ecnt]=v,w[ecnt]=val;
	nxt[++ecnt]=last[v],last[v]=ecnt,to[ecnt]=u,w[ecnt]=val;
}

struct data{
	ll v;
	int k;
	data(){}
	data(cs ll &_v,cs int &_k):v(_v),k(_k){}
	friend bool operator<(cs data &a,cs data &b){
		return a.v==b.v?a.k>b.k:a.v<b.v;
	}
	friend data operator+(cs data &a,cs data &b){
		return data(a.v+b.v,a.k+b.k);
	}
}dp[N][3],tr[3];

cs ll lINF=0x7f7f7f7f7f;
cs int INF=0x3f3f3f3f;

ll mid;

inline void dfs(int u,int fa){
	for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]])
	if(v^fa){
		dfs(v,u);
		for(int re i=0;i<3;++i)tr[i]=data(-lINF,INF);
		for(int re i=0;i<3;++i)tr[0]=max(tr[0],dp[u][0]+dp[v][i]);
		tr[1]=max(tr[1],dp[u][0]+dp[v][0]+data(w[e]-mid,1));
		tr[1]=max(tr[1],dp[u][0]+dp[v][1]+data(w[e],0));
		for(int re i=0;i<3;++i)tr[1]=max(tr[1],dp[u][1]+dp[v][i]);
		tr[2]=max(tr[2],dp[u][1]+dp[v][0]+data(w[e],0));
		tr[2]=max(tr[2],dp[u][1]+dp[v][1]+data(w[e]+mid,-1));
		for(int re i=0;i<3;++i)tr[2]=max(tr[2],dp[u][2]+dp[v][i]);
		for(int re i=0;i<3;++i)dp[u][i]=tr[i];
	}
}

int n,k;
inline void init(){
	for(int re i=1;i<=n;++i){
		dp[i][0]=data(0,0);
		dp[i][1]=data(-lINF,INF);
		dp[i][2]=data(-mid,1);
	}
}

ll l,r;
data res;
signed main(){
	n=getint();
	k=getint();
	for(int re i=1,u,v,val;i<n;++i){
		u=getint(),v=getint(),val=getint();
		addedge(u,v,val);
		r+=abs(val);
	}
	l=-r;
	while(l<=r){
		mid=(l+r)/2;
		init();
		dfs(1,0);
		res=max(dp[1][0],max(dp[1][1],dp[1][2]));
		if(res.k==k+1){
			cout<<res.v+(k+1)*mid;
			return 0;
		}
		if(res.k>k+1)l=mid+1;
		else r=mid-1;
	}
	mid=l;
	init();
	dfs(1,0);
	res=max(dp[1][0],max(dp[1][1],dp[1][2]));
	cout<<res.v+(k+1)*mid;
	return 0;
}