Find Metal Mineral

老和山下的小学童 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;
}

相关推荐