hdu33339 In Action

基于01背包与Floyd算法解决电站破坏问题

最新推荐文章于 2020-03-06 16:56:36 发布

转载最新推荐文章于 2020-03-06 16:56:36 发布 · 641 阅读

文章标签：

#最短路径

最短路径专栏收录该内容

25 篇文章

订阅专栏

本文探讨了使用01背包算法和Floyd算法解决从多个起点到多个终点破坏电站问题的策略，通过计算最短路径和背包问题的结合，实现了最优解的求解。

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

using namespace std;
/*
理解题意很重要， 一次从0 点开始到i， 破坏一个电站， 再从0 开始到j 破坏第二个，
 思路； 最段路径 + 01 背包；

*/

const int M = 105;
const int N = 10005;
const int MAX = 999999;
int dist[M];
int cost[M];
int weight[M];
int mmap[M][M];
int dp[N];
int T, n, m;
int sum, s;

void init() {

    for(int i = 0; i <= n; i++) {

        for(int j = 0; j <= n; j++) {
            mmap[i][j] = MAX;
        }
    }

}

void floyd(int n) {

   for(int k = 0; k <= n; k++) {

    for(int i = 0; i <= n; i++) {

        for(int j = 0; j <= n; j++) {

            mmap[i][j] = min(mmap[i][j], mmap[i][k] + mmap[k][j]);
        }
    }
   }
}

void zeroPake(int s, int n) {

    memset(dp, 0, sizeof(dp));

    for(int i = 1; i <= n; i++) {

       for(int j = s; j >= cost[i]; j--) {

          dp[j] = max(dp[j], dp[j - cost[i]] + weight[i]);
       }
    }
}

int main()
{
    int x, y, d;
    scanf("%d", &T);
    while(T--) {


        scanf("%d%d", &n, &m);
         init();

        for(int i = 0; i < m; i++) {

            scanf("%d%d%d", &x, &y, &d);

            if(mmap[x][y] > d) {

                mmap[x][y] = mmap[y][x] = d;
            }
        }
            sum = 0;
            for(int i = 1; i <= n; i++) {

                scanf("%d", &weight[i]);
                sum += weight[i];
            }
            sum = (sum>>1) + 1;
            floyd(n);

            s = 0;

            for(int i = 1; i <= n; i++) {

                cost[i] = mmap[0][i];
                if(cost[i] != MAX) {

                    s += cost[i];
                }
            }

            zeroPake(s, n);
          int flag = 0;
          for(int i = 1; i <= s; i++) {

            if(dp[i] >= sum) {

                flag = 1;
                printf("%d\n", i);
                break;
            }
          }
          if(!flag)
            printf("impossible\n");

    }
    return 0;
}