2020.08.03日常总结

二分是个神奇的算法，它能让你难以维护的数据突然变得十分容易维护……

$$\text{洛谷P4322 [JSOI2016]最佳团体}$$

$\text{[Problem]}$

• 有一个有 $N+1$ 个点的树，每个节点有两个值：$p_i,s_i(1\le i\le N)$。

• 你需要选 $t+1$ 个点，如果一个点选了，则其父亲必须要选。

• 一个选择方案集合 $T$ 的总价值为：

  $$\frac{\sum _{i\in T}{p}_i}{\sum _{i\in T}{s}_i}$$

• 求总价值的最大值。$1\le t\le N\le 2500,1\le p_i,s_i\le 1\times 10^4(1\le i\le N),\text{Father}(i)<i$。节点从 $0$ 编号到 $N$。

$\text{[Solution]}$

是不是感觉总价值的那个式子特别难维护？

我们可以记 $\text{mid}$ 表示总价值大于等于 $\text{mid}$。我们发现 $\text{mid}$ 越大，越难以满足，所以我们可以二分答案。

改写一个总价值那个式子：

$$\begin{array}{rl}\frac{\sum _{i\in T}{p}_i}{\sum _{i\in T}{s}_i}& \ge \text{mid}\\ \sum _{i\in T}{p}_i& \ge \text{mid}\times \sum _{i\in T}{s}_i\\ \sum _{i\in T}{p}_i-\text{mid}\times \sum _{i\in T}{s}_i& \ge 0\\ \sum _{i\in T}\left(p_i-\text{mid}\times s_i\right)& \ge 0\end{array}$$

改每个点的价值为 $p_i-\text{mid}\times s_i$，原题被转化为了一个树上 dp 的问题。只要最后的总价值 $\ge 0$，代表 $\text{mid}$ 是一个可行答案。

如何树上 dp？很简单，记 $f_{i,j}$ 表示在 $i$ 的子树中选 $j$ 个点时的最大总价值，原题被转化为了一个树上背包问题，可以用类似 选课 一题的方法解决。

$\text{[code]}$

struct edge{//链式前向星 
	int next,to;//的模板 
}e[2510];int h[2510],tot;
inline void add(int a,int b){
	e[++tot]=(edge){h[a],b};h[a]=tot;
}
double p[2510],s[2510],w[2510];
double f[2510][2510];//dp用数组 
int n,t,sze[2510];double l,r,mid;
inline void dp(int u){
	f[u][1]=w[u];sze[u]=1;//init
	for(int i=h[u];i;i=e[i].next){
		register int to=e[i].to;
		dp(to);//先递归计算儿子值 
		sze[u]+=sze[to];//子树大小 
		for(int k=min(sze[u],t+1);k>=1;k--)
			for(int l=0;l<=min(sze[to],k-1);l++)
				f[u][k]=max(f[u][k],f[u][k-l]+f[to][l]);
//				该状态转移方程的含义是：在u当前加入的所有儿子中，to这个子树选l个点，其它儿子所对应的子树共选k-l个点时的最大总价值
//				注意就像01背包一样,k必须是倒序的
	}
}
const int inf=0x3f3f3f3f;
inline bool check(double mid){
	for(int i=1;i<=n;i++)
		w[i]=p[i]-mid*s[i];
	w[0]=0;//计算每个人贡献 
	for(int i=0;i<=n;i++)
		for(int j=1;j<=t+1;j++)
			f[i][j]=-inf;
	dp(0);//从根开始进行dp 
	return f[0][t+1]>=0;
}
const double eps=1e-4;
int main(){
	t=read();n=read();
	for(int i=1;i<=n;i++){
		s[i]=read();p[i]=read();
		add(read(),i);//建图 
	}
	l=0.0;r=10000.0;
	while (l+eps<r){
		mid=(l+r)/2.0;
		if (check(mid)) l=mid;
		else r=mid;
	}
	printf("%.3lf",l);
	return 0;
}