POJ 2152 Fire

题目传送门

题目翻译:

描述

Z国有N个城市，编号从1到N。城市之间通过高速公路相连，并且两个不同城市之间只有一条路径。Z国最近经常起火，因此政府决定在一些城市建造一些消防站。在城市K成本W（K）中建立一个消防站。不同城市的W可能不同。如果城市K中没有消防站，那么它与最近的拥有消防站的城市之间的距离不能超过D（K）。不同城市的D也可能不同。为了省钱，政府希望您计算建造消防站的最低费用。

输入值

输入的第一行包含一个表示测试用例数量的整数T。接下来的T块分别代表一个测试案例。 

每个块的第一行包含一个整数N（1 <N <= 1000）。第二行包含由一个或多个空格分隔的N个数字。第I个数字表示W（I）（0 <W（I）<= 10000）。第三行包含由一个或多个空格分隔的N个数字。第I个数字表示D（I）（0 <= D（I）<= 10000）。接下来的N-1行分别包含三个整数u，v，L（1 <= u，v <= N，0 <L <= 1000），这意味着在城市u和v之间有一条长度为L的高速公路。 

输出量

对于每个测试用例，单行输出的最低成本。

样本输入

5
5
1 1 1 1 1
1 1 1 1 1
1 2 1
2 3 1
3 4 1
4 5 1
5
1 1 1 1 1
2 1 1 1 2
1 2 1
2 3 1
3 4 1
4 5 1
5
1 1 3 1 1
2 1 1 1 2
1 2 1
2 3 1
3 4 1
4 5 1
4
2 1 1 1
3 4 3 2
1 2 3
1 3 3
1 4 2
4
4 1 1 1
3 4 3 2
1 2 3
1 3 3
1 4 2

样本输出

2
1
2
2
3

解题思路:

实在太复杂,不会描述,看大佬博客https://www.cnblogs.com/qzqzgfy/p/5553912.html

AC代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

int best[1001],dist[1001],t,w[1001],n,f[1001][1001],d[1001],head[1001],tot;
struct kkk {
    int to,v,next;
}e[2001];

inline void add(int x,int y,int z) {
    e[++tot].to = y;
    e[tot].v = z;
    e[tot].next = head[x];
    head[x] = tot;
}

inline void Distance(int root) {
    for(int i = head[root];i != 0; i = e[i].next) {
        int u = e[i].to;
        if(dist[u] != -1) continue;
        dist[u] = dist[root] + e[i].v;
        Distance(u);    
    }
}

inline void dfs(int root,int fa) {
    for(int i = head[root];i != 0; i = e[i].next) {
        int u = e[i].to;
        if(fa == u) continue;
        dfs(u,root);
    }
    for(int i = 1;i <= n; i++)
        dist[i] = -1;//为了不让自己更新自己,对答案无影响 
    dist[root] = 0;
    Distance(root);
    best[root] = 0x3f3f3f3f3f;
    for(int i = 1;i <= n; i++) f[root][i] = 0x3f3f3f3f;
    for(int i = 1;i <= n; i++)
        if(dist[i] <= d[root]) {
            f[root][i] = w[i];
            for(int j = head[root];j != 0; j = e[j].next) {
                int u = e[j].to;
                if(u == fa) continue;
                f[root][i] += min(best[u],f[u][i] - w[i]);
            }
            best[root] = min(best[root],f[root][i]);
        }
}

int main() {
    scanf("%d",&t);
    while(t--) {
        memset(head,0,sizeof(head));
        tot = 0;
        scanf("%d",&n);
        for(int i = 1;i <= n; i++)    
            scanf("%d",&w[i]);
        for(int i = 1;i <= n; i++)
            scanf("%d",&d[i]);
        for(int i = 1;i < n; i++) {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            add(x,y,z);
            add(y,x,z);
        }
        dfs(1,-1);
        printf("%d
",best[1]);
    }
    return 0;
}