老和山下的小学童 2020-02-23
题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4003
题意:
给一棵n个节点的树, 节点编号为1~n, 每条边都有一个花费值.
有k个机器人从S点出发, 问让机器人遍历所有边,最少花费值多少?
思路:
用f(i, j)表示子树i用j个机器人的最少花费
如果从根节点出发,遍历所有节点之后再回到原点, 那么最少的花费一定是所有边的权值之和sum的两倍, 因为每条边都走了两次.
而这题, 遍历完之后,并不需要走回出发点, 所以, 有些边只走了一次就可以了,
如果用1台机器人走, 最少的的花费 = sum * 2 - {根节点到叶子节点路径的最大权值和}
如果是j台机器走, 我们要让j台机器人只走一次的边的权值之和尽量大, 也就是减少的花费尽量大.
那么, 我的状态表示为:
f(i, j) 表示子树i用j个机器人最多可以减少的花费.
对于i节点, 它的每个子节点的子树是一组物品, 我们可以选择派1,2,...j个机器人走去
需要注意, 如果派x个机器人走向某个子节点v, 那么边edge(i, v)就会被走了x次, 花费了x*w(i, v).
而原始的sum中每条边只走了两次, 所以走edge(i, v)的花费减少了 2*w(i,v) - x*w(i,v)
最后可以得到状态转移式:
f(i, j) = max{ max{f(i, j-k) + f(v, k) + 2*w(i,v) - k*w(i,v) | 1<=k<=j } | v是i的儿子节点 }
#include <iostream> #include <algorithm> #include <string> #include <string.h> #include <vector> #include <map> #include <stack> #include <set> #include <queue> #include <math.h> #include <cstdio> #include <iomanip> #include <time.h> #include <bitset> #include <cmath> #define LL long long #define INF 0x3f3f3f3f #define ls nod<<1 #define rs (nod<<1)+1 const double eps = 1e-10; const int maxn = 1e4 + 10; const LL mod = 1e9 + 7; int sgn(double a){return a < -eps ? -1 : a < eps ? 0 : 1;} using namespace std; struct edge { int v,w,nxt; }e[maxn<<1]; int head[maxn]; int cnt; int f[maxn][15]; int n,m,k; inline void add_edge(int u,int v,int w) { e[++cnt].v = v; e[cnt].w = w; e[cnt].nxt = head[u]; head[u] = cnt; } inline void dfs(int x,int fa) { for (int i = head[x];~i;i = e[i].nxt) { int v = e[i].v; if (v == fa) continue; dfs(v,x); for (int j = k;j >= 1;j--) { for (int l = 1;l <= j;l++) { f[x][j] = max(f[x][j],f[x][j-l]+f[v][l]+2*e[i].w-l*e[i].w); } } } } int main() { ios::sync_with_stdio(false); while (cin >> n >> m >> k) { cnt = 0; memset(head,-1, sizeof(head)); memset(f,0, sizeof(f)); int sum = 0; for (int i = 1;i < n;i++) { int u,v,w; cin >> u >> v >> w; sum += w; add_edge(u,v,w); add_edge(v,u,w); } dfs(m,0); cout << 2*sum - f[m][k] << endl; } return 0; }