退役选手的编程挑战：最小生成树与概率优化算法应用-CSDN博客

本文链接：https://blog.csdn.net/oi_521289/article/details/121259578

都是要退役的人了，不知道为啥想学，就看了看，做了两个板子，写个笔记。

例一 P2619 [国家集训队]Tree I

很早以前就做过了，但根本就搞不明白为什么要去调整白边的权值二分答案。

简要题意

有一个 $n$ 个点 $m$ 条边的无向图，其中每一条边都有权值、颜色，问恰好有 $n e e d$ 条白边的最小生成树的答案。 $(n < = 5 e 4, n e e d < = m < = 1 e 5)$

题解

设 $g (x)$ 表示恰好选 $x$ 条白边的最小生成树的答案，那么答案就是 $g (n e e d)$ 。但是直接求解显然复杂 ~~（我不会）~~ ，考虑 $g$ 函数，我们就是要求这个函数 $x = n e e d$ 时的 $y$ 值。发现 $g$ 是凹函数，也就是这个函数的斜率 $k$ 单增，考虑去二分 $k$ 值，找到斜率 $k$ 切的点的位置，即满足 $g (x) = k x + b$ ，发现一定是让截距 $b$ 取 $m i n$ 的位置 $x$ 。又因为 $b = g (x) - k x$ ，即问题变成找 $x$ 让 $g (x) - k x$ 最小，这道题中 $g (x)$ 的 $x$ 的含义就是选 $x$ 条白边时的总边权的最小，由于每选一条白边影响都是 $- k$ 是一样的，那把每条白边的权值 $- k$ 求最小生成树就能知道 $x$ 的取值了，即此时选择白边的数量。然后比较 $x$ 与 $n e e d$ 就能知道当前斜率 $k$ 偏大/偏小，最后二分出 $k$ 再跑遍最小生成树就能得到答案（注意要 $+ k * n e e d$ 补充没有考虑的）。注意题目条件 $a b s (边权) < = 100$ ，斜率就在 $- 100$ 到 $100$ 之间二分就可以了。

#include<bits/stdc++.h>
using namespace std;

const int N=5e4+10,M=1e5+10,K=110;
int n,m,x,y,z,need,fa[N];
struct Edge{
	int x,y,v;
	inline bool operator < (const Edge &b)const{
		return v<b.v;
	}
}e1[M],e2[M]; int cnt1,cnt2;

inline int getfa(const int x){
	return x==fa[x]?x:fa[x]=getfa(fa[x]);
}
inline bool merge(const int x,const int y){
	int fx=getfa(x),fy=getfa(y); if(fx==fy) return 0;
	fa[fx]=fy; return 1;
}
inline bool check(const int k){
	for(int i=0;i<n;i++) fa[i]=i; int use=0;
	int i1=1,i2=1; while(i1<=cnt1&&i2<=cnt2)
		(e1[i1].v<e2[i2].v-k)?(merge(e1[i1].x,e1[i1].y),++i1):(use+=merge(e2[i2].x,e2[i2].y),++i2);
	while(i2<=cnt2) use+=merge(e2[i2].x,e2[i2].y),++i2;
	return use<need;
}
#define gc getchar()
#define in read()
inline int read(){
	int f=1,k=0; char cp=gc; while(cp!='-'&&(cp<'0'||cp>'9')) cp=gc; if(cp=='-') cp=gc;
	while(cp>='0'&&cp<='9') k=(k<<3)+(k<<1)+cp-48,cp=gc; return f*k;
}
int main(){ 
	n=in,m=in,need=in;
	for(int i=1;i<=m;i++){ x=in,y=in,z=in;
		if(in) e1[++cnt1]=Edge{x,y,z}; 
		else e2[++cnt2]=Edge{x,y,z};
	} sort(e1+1,e1+cnt1+1),sort(e2+1,e2+cnt2+1);
	int l=-K,r=K; while(l<r){ int mid=(l+r)>>1;
		if(check(mid)) l=mid+1; else r=mid;
	} int k=l; 
	for(int i=0;i<n;i++) fa[i]=i; int ans=0;
	int i1=1,i2=1; while(i1<=cnt1&&i2<=cnt2)
		(e1[i1].v<e2[i2].v-k)?(ans+=e1[i1].v*merge(e1[i1].x,e1[i1].y),++i1):(ans+=(e2[i2].v-k)*merge(e2[i2].x,e2[i2].y),++i2);
	while(i2<=cnt2) ans+=(e2[i2].v-k)*merge(e2[i2].x,e2[i2].y),++i2;
	while(i1<=cnt1) ans+=e1[i1].v*merge(e1[i1].x,e1[i1].y),++i1;
	printf("%d\n",ans+k*need);

	return 0;
}

例二 CF739E Gosha is hunting

简要题意（由概率期望魔性翻译而来）

有n个物品，分别有两种价值 $a_1,a_2,...,a_n$ 和 $b_1,b_2,...,b_n$ ，给定两种价值分别可以选的个数为 $A, B$ ，可以任选并收益对应的价值，但是对于同一个 $i$ ，如果两种价值都选了则收益减少 $a_i\times b_i$ ，求最大收益。 $(A, B < = n < = 2000)$

题解

$n^3$ 做法显然， $f [k] [i] [j]$ 表示前 $k$ 个选了 $i$ 个价值一、 $j$ 个价值二的最大收益。考虑优化，如果固定了 $i$ ，那对于 $f [n] [i] [x]$ 来说一定是关于 $x$ 的一个凸函数，就可以用 $w q s$ 二分求出价值一个数 $i$ 对应应选出价值二个数 $j$ 的数量和其对于的最大收益，复杂度变成 $O(n^2logk)$ 。考虑价值一也满足这个性质，所以变成 $w q s$ 二分套 $w q s$ 二分，复杂度优化到 $O(nlog^2k)$ ，可以通过本题。

#include<bits/stdc++.h>
using namespace std;

const int N=2010;
const double eps=1e-8;
int n,a,b,numa[N],numb[N]; double pa[N],pb[N],f[N],ka,kb;

inline void check_b(const double ka,const double kb){
	memset(numa,0,sizeof(numa)),memset(numb,0,sizeof(numb)),memset(f,0,sizeof(f));
	for(int i=1;i<=n;i++){
		f[i]=f[i-1],numa[i]=numa[i-1],numb[i]=numb[i-1];
		if(f[i]<f[i-1]+pa[i]-ka) f[i]=f[i-1]+pa[i]-ka,numa[i]=numa[i-1]+1,numb[i]=numb[i-1];
		if(f[i]<f[i-1]+pb[i]-kb) f[i]=f[i-1]+pb[i]-kb,numb[i]=numb[i-1]+1,numa[i]=numa[i-1];
		if(f[i]<f[i-1]+pa[i]+pb[i]-pa[i]*pb[i]-ka-kb) //注意ka,kb的实际含义是选一次a,b所需要的代价 
			f[i]=f[i-1]+pa[i]+pb[i]-pa[i]*pb[i]-ka-kb,numa[i]=numa[i-1]+1,numb[i]=numb[i-1]+1;
	}
}
inline bool check_a(const double ka){
	double l=0,r=1; for(int i=1;i<=50;i++){ 
		double mid=(l+r)/2.0; check_b(ka,mid);
		if(numb[n]<b) r=mid-eps; else l=mid;
	} kb=l; check_b(ka,kb); return numa[n]<a;
}
#define gc getchar()
#define in read()
inline int read(){
	int f=1,k=0; char cp=gc; while(cp!='-'&&(cp<'0'||cp>'9')) cp=gc; if(cp=='-') cp=gc;
	while(cp>='0'&&cp<='9') k=(k<<3)+(k<<1)+cp-48,cp=gc; return f*k;
}
int main(){
	n=in,a=in,b=in; 
	for(int i=1;i<=n;i++) scanf("%lf",&pa[i]);
	for(int i=1;i<=n;i++) scanf("%lf",&pb[i]);
	double l=0,r=1; for(int i=1;i<=50;i++){ //满足精度就行 
		double mid=(l+r)/2.0; if(check_a(mid)) r=mid-eps; else l=mid;
	} ka=l; check_a(ka); printf("%.7lf\n",f[n]+ka*a+kb*b); //yyb说最好多保留3位

	return 0;
}