P1642 规划

P1642 规划

题意：

某地方有N个工厂，有N-1条路连接它们，且它们两两都可达。每个工厂都有一个产量值和一个污染值。现在工厂要进行规划，拆除其中的M个工厂，使得剩下的工厂依然连成一片且 总产量/总污染 的值最大。

题解：

01分数规划+树形dp
01分数规划这里就不细讲了，详细看这里
对我来说难的是树形dp（捂脸）
本题要求所选的工厂依然可以连成一片，我一开始想的是直接dfs判断，但是可以判断是否连通，但是无法求出最大值来
有什么东西既可以保证连通又可以求出最大值
额，树形dp
我们设dp[u][j]表示以u为根的子树中，有一个点数位j的连通块时的最大值。这个连通块一定是包含u的
转移方程：
f[x][j] = max(f[x][j], f[x][j - k] + f[v][k]);
v是x的儿子

代码：

#include<bits/stdc++.h>
#define N 1000005
#define INF 23333333
using namespace std;
const double eps = 0.000001;
struct edge{	
	int v, nxt;
}e[N];
int p[N], eid;
void init(){
	memset(p, -1, sizeof p);
	eid = 0;
}
void insert(int u, int v){
	e[eid].v = v;
	e[eid].nxt = p[u];
	p[u] = eid ++;
}
double f[105][105], d[105];
int vis[105], size[105], n, m, a[105], b[105];
void dfs(int x){//简单的树dp， f[i][j]表示第 i 个点， 选 j 个的方案数， 其中 i 必须选 
	vis[x] = 1;
	size[x] = 1;
	f[x][1] = d[x];
	f[x][0] = 0;
	for(int i = p[x]; i + 1; i = e[i].nxt){
		int v = e[i].v;
		if(vis[v]) continue;
		dfs(v); 
		size[x] += size[v];
		for(int j = min(m, size[x]); j >= 1; j --){
			for(int k = 0; k < j; k ++){
				f[x][j] = max(f[x][j], f[x][j - k] + f[v][k]);
			}
		}
	}
}
int check(double x){
	for(int i = 1; i <= n; i ++) 
	{
		d[i] = a[i] * 1.0 - b[i] * 1.0 * x;
		vis[i] = 0;
		size[i] = 0; //求 d[i] 	
	}
	for(int i = 1; i <= n; i ++)
		for(int j = 0; j <= m; j ++) 
			f[i][j] = -INF;
	dfs(1);
	for(int i = 1; i <= n; i ++){
		if(f[i][m] > -eps) return 1; //如果有满足要求的就退出 
	}
	return 0;
}
int main(){
	init();
	scanf("%d%d", &n, &m);
	m = n - m;
	for(int i = 1; i <= n; i ++) scanf("%d", &a[i]);
	for(int i = 1; i <= n; i ++) scanf("%d", &b[i]);
	
	for(int i = 1; i < n; i ++){
		int u, v;
		scanf("%d%d", &u, &v);
		insert(u, v);
		insert(v, u);
	}	
	double l = 0, r = 100000;
	while(l + eps < r){ //二分答案 
		double mid = (l + r) * 1.0 / 2.0;
		if(check(mid)) l = mid;
		else r = mid;
	}
	printf("%.1f", l);
	return 0;
}