洛谷.4383.[八省联考2018]林克卡特树lct(树形DP 带权二分)

题目链接

\(Description\)

给定一棵边带权的树。求删掉K条边、再连上K条权为0的边后，新树的最大直径。
\(n,K\leq3\times10^5\)。

\(Solution\)

题目可以转化为，求树上不相交的\(k+1\)条链，使得它们的边权和最大(已不想再说什么了。。)。
选择链数越多，答案增长得越慢，减少的时候还会减少得越快，即形成了一个\(K-Ans_K\)的上凸包；而如果没有链数的限制，DP是很容易的(有链数得加一维\(k\))。
带权二分。DP用\(f[x][0/1/2]\)表示点\(x\)度数为\(0/1/2\)时的最优解，记一下最优情况下的链数。

DP细节：
\(f[x][1]\)即度数为\(1\)时不加作为链的花费，而是合并时加上，更方便吧。
最后用\(f[x][0]\) 与 以\(f[x][1]\)结束链或是\(f[x][2]\)取个\(\max\)，表示最终状态（不再向上更新的最优状态，即从这断开）。
结构体写虽然可能慢点但是太好写了。但常数竟然这么大的么...
注意是\(K+1\)→_→

[Update] 19.2.11
二分边界是，使得边界足够大能保证每一个物品都不会选，也就是每个物品的最大可能值就可以了。（比如CF739E，权值0~1就够）
然后...二分的时候只要保证恰好取到\(k\)个就可以了，斜率具体是多少无所谓...吧。

//22156ms	31.88MB
#include <cstdio>
#include <cctype>
#include <algorithm>
//#define gc() getchar()
#define MAXIN 200000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=3e5+5;

int n,K,Enum,H[N],nxt[N<<1],to[N<<1],len[N<<1];
char IN[MAXIN],*SS=IN,*TT=IN;
LL C,sum;
struct Node{
	LL v; int n;
	Node() {}
	Node(LL v,int n):v(v),n(n) {}
	bool operator <(const Node &x)const{
		return v==x.v?n>x.n:v<x.v;
	}
	Node operator +(const Node &x){
		return Node(v+x.v, n+x.n);
	}
	Node operator +(LL val){
		return Node(v+val, n);
	}
}f[N][3];

inline int read()
{
	int now=0,f=1;register char c=gc();
	for(;!isdigit(c);c=gc()) if(c=='-') f=-1;
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now*f;
}
inline void AddEdge(int u,int v)
{
	int w=read(); sum+=abs(w);
	to[++Enum]=v, nxt[Enum]=H[u], len[Enum]=w, H[u]=Enum;
	to[++Enum]=u, nxt[Enum]=H[v], len[Enum]=w, H[v]=Enum;	
}
inline Node Update(Node t){//合并成一条整链 
	return Node(t.v-C, t.n+1);
}
void DFS(int x,int fa)
{
	f[x][0]=f[x][1]=Node(0,0), f[x][2]=Node(-C,1);
	//但是最初f[x][1/2]不应该没有值吗。。但是这样初始化没问题 因为如果只是这种情况也不会比f[x][0]更优吧。
	for(int v,val,i=H[x]; i; i=nxt[i])
		if((v=to[i])!=fa)
		{
			DFS(v,x), val=len[i];
			f[x][2]=std::max(f[x][2]+f[v][0],Update(f[x][1]+f[v][1]+val));
			f[x][1]=std::max(f[x][1]+f[v][0],f[x][0]+f[v][1]+val);
			f[x][0]=f[x][0]+f[v][0];
		}
	f[x][0]=std::max(f[x][0],std::max(Update(f[x][1]),f[x][2]));
}

int main()
{
	n=read(), K=read()+1;
	for(int i=1; i<n; ++i) AddEdge(read(),read());
	LL l=-sum,r=sum;
	while(l<=r)
	{
		if(C=l+r>>1, DFS(1,1), f[1][0].n>K) l=C+1;
		else r=C-1;
	}
	C=l, DFS(1,1);//最后以l(r+1)为答案。
	printf("%lld",f[1][0].v+K*l);

	return 0;
}

新写的代码：（差不多...）

#include <cstdio>
#include <cctype>
#include <algorithm>
#define gc() getchar()
typedef long long LL;
const int N=3e5+5;
const LL INF=1ll<<60;

int Enum,H[N],nxt[N<<1],to[N<<1],len[N<<1];
LL C;
struct Node
{
	LL val; int cnt;
	inline Node operator +(int v)
	{
		return (Node){val+v,cnt};
	}
	inline Node operator +(const Node &x)
	{
		return (Node){val+x.val,cnt+x.cnt};
	}
	inline bool operator <(const Node &x)const
	{
		return val==x.val?cnt>x.cnt:val<x.val;
	}
}f[N][3];

inline int read()
{
	int now=0,f=1;register char c=gc();
	for(;!isdigit(c);c=='-'&&(f=-1),c=gc());
	for(;isdigit(c);now=now*10+c-48,c=gc());
	return now*f;
}
inline void AE(int u,int v,int w)
{
	to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum, len[Enum]=w;
	to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum, len[Enum]=w;
}
inline Node Upd(const Node &x)
{
	return (Node){x.val-C,x.cnt+1};
}
void DFS(int x,int fa)
{
	f[x][0]=f[x][1]=(Node){0,0}, f[x][2]=(Node){-INF,0};
	//f[x][1]=0，直接合并。
	for(int i=H[x],v; i; i=nxt[i])
		if((v=to[i])!=fa)
		{
			DFS(v,x);
			f[x][2]=std::max(f[x][2]+f[v][0],Upd(f[x][1]+f[v][1]+len[i]));
			f[x][1]=std::max(f[x][1]+f[v][0],f[x][0]+f[v][1]+len[i]);
			f[x][0]=f[x][0]+f[v][0];
		}
	f[x][0]=std::max(f[x][0],std::max(Upd(f[x][1]),f[x][2]));//为方便直接把f[x][0]作为在x处断开的最优值即可。
}

int main()
{
	freopen("lct.in","r",stdin);
	freopen("lct.out","w",stdout);

	const int n=read(),K=read()+1;
	LL s1=0,s2=0;
	for(int i=1,u,v,w; i<n; ++i) u=read(),v=read(),w=read(),w>0?s1+=w:s2-=w,AE(u,v,w);
	LL r=std::max(s1,s2),l=-r,mid;
	while(l<r)
	{
		if(C=mid=l+r>>1,DFS(1,1),f[1][0].cnt>K) l=mid+1;
		else r=mid;
	}
	C=l, DFS(1,1);
	printf("%lld\n",f[1][0].val+C*K);

	return 0;
}