洛谷P1273 有线电视网(DP,分组背包)
题目描述
某收费有线电视网计划转播一场重要的足球比赛。他们的转播网和用户终端构成一棵树状结构,这棵树的根结点位于足球比赛的现场,树叶为各个用户终端,其他中转站为该树的内部节点。从转播站到转播站以及从转播站到所有用户终端的信号传输费用都是已知的,一场转播的总费用等于传输信号的费用总和。
现在每个用户都准备了一笔费用想观看这场精彩的足球比赛,有线电视网有权决定给哪些用户提供信号而不给哪些用户提供信号。
写一个程序找出一个方案使得有线电视网在不亏本的情况下使观看转播的用户尽可能多。
输入输出格式
输入格式:
输入文件的第一行包含两个用空格隔开的整数N和M,其中2≤N≤3000,1≤M≤N-1,N为整个有线电视网的结点总数,M为用户终端的数量。第一个转播站即树的根结点编号为1,其他的转播站编号为2到N-M,用户终端编号为N-M+1到N。
接下来的N-M行每行表示—个转播站的数据,第i+1行表示第i个转播站的数据,其格式如下:
K A1 C1 A2 C2 … Ak Ck
K表示该转播站下接K个结点(转播站或用户),每个结点对应一对整数A与C,A表示结点编号,C表示从当前转播站传输信号到结点A的费用。最后一行依次表示所有用户为观看比赛而准备支付的钱数。
输出格式:
输出文件仅一行,包含一个整数,表示上述问题所要求的最大用户数。输入输出样例
输入样例#1:
5 32 2 2 5 3
2 3 2 4 3
3 4 2
输出样例#1:
2说明
样例解释2+3+2+3=10,大于用户愿意支付的总费用3+4+2=9,有线电视网就亏本了,而只让③④两个用户看比赛就不亏本了。
解题分析
对于有线电视公司,客户k的盈利为:其所支付的money-他到根节点这一路径上所有转播站的费用,但如果其兄弟结点一起算的话,他们到根节点的公共路径上的转播站的费用只需要算一次。这就导致该问题不能只计算单个客户的盈利。因此,一个结点的上的子树必须整体考虑,即考虑取1个客户、2个客户、....,因为不管在该节点的子树上去多少个客户,从该结点到根节点之间的总费用都是一样的。
这样就得到状态状态:f(i, j),表示以结点i为根节点的子树中,取j个客户的最大盈利数。
对于结点i,假设其孩子结点有k个,每个孩子结点的客户数为c1, c2, ..., ck,因此在每个孩子结点(例如最左孩子)有可能取且只可能去下列情形之一:0个、1个、两个、...c1个,这是典型的分组背包问题。
状态转移方程:f(i, j) = max(f(i, j), f(i, j-x) + f(y, x) - cost(y, i)。
其中,y为i的一个孩子结点,x为y为根节点的子树中可能取到的客户数, cost(y, i)为i向y转播信号的花费。
初始化:对于客户结点(叶结点), 有f(i, 1) = money(i),这里,money(i)为客户i所缴纳的费用。对于其他所有情况,f(i, j)都设置为一个很小的负数。
可以用最左孩子和右兄弟的方式表示图,在每一个结点,需要有两个量:父节点到它的费用以及以该节点为根节点的子树中客户的数量。
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std; #define N 3002 #define INF 0x77ffffff #define MAX(a, b) ((a)>(b)?(a):(b)) int n, m, ans = 0, profit = 0, f[N][N]; // f[i][j]:以i为根节点的子树中客户数为j的最大盈利 struct node { int leftmostchild; // 最左孩子 int rightsibling; // 右兄弟 int cost; // 从父节点都该节点的费用 int customs; // 以该结点为根节点的用户的数量 node():leftmostchild(-1),rightsibling(-1),cost(0), customs(0) { } } tree[N]; void get_i(int &x) { char ch = getchar(); x = 0; while(!isdigit(ch)) ch = getchar(); while(isdigit(ch)) { x = x * 10 + ch - '0'; ch = getchar(); } } void dfs(int st) { int now, i, j, num; now = tree[st].leftmostchild; while(~now) { dfs(now); tree[st].customs += tree[now].customs; now = tree[now].rightsibling; } // 分组背包 now = tree[st].leftmostchild; num = 0; while(~now) { num+=tree[now].customs; for(i=num; i>0; i--) for(j=0; j<=i; j++) f[st][i] = MAX(f[st][i], f[st][i-j]+f[now][j] - tree[now].cost); now = tree[now].rightsibling; } } int main() { ios::sync_with_stdio(false); int i, j, k, a, c, a1; get_i(n), get_i(m); for(i=1; i<=n-m; i++) { get_i(k); get_i(a), get_i(c); tree[i].leftmostchild = a; tree[a].cost = c; a1 = a; for(j=1; j
0; i--) if(f[1][i]>=0) { cout<
<