http://acm.hdu.edu.cn/showproblem.php?pid=3339
题意:有n个基地,每个基地都有一定的能量,到达每个基地需要一定的花费,tank从0出发,每个基地都有一个tank,选择一些基地使得他们的能量和大于总能量的一半且花费最少。
思路:
先floyd求出0到每个基地的耗油量,即花费cost[i]。
以cost[i]为花费,weight[i]为价值进行01背包,每个基地有选择和不选择两种。
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
int n,m;
int sump,sumv;
int dis[110][110];
int cost[110],weight[110];
int dp[1000010];
void floyd()
{
for(int k = 0; k <= n; k++)
{
for(int i = 0; i <= n; i++)
{
for(int j = 0; j <= n; j++)
dis[i][j] = min(dis[i][j],dis[i][k]+dis[k][j]);
}
}
}
//01背包,以基地的能量为价值,耗油量为花费
void ZeroOnePack()
{
memset(dp,0,sizeof(dp));
for(int i = 1; i <= n; i++)
{
for(int j = sumv; j >= cost[i]; j--)
dp[j] = max(dp[j],dp[j-cost[i]]+weight[i]);
}
}
int main()
{
int test;
scanf("%d",&test);
while(test--)
{
scanf("%d %d",&n,&m);
for(int i = 0; i <= n; i++)
{
for(int j = 0; j <= n; j++)
{
if(i == j)
dis[i][j] = 0;
else
dis[i][j] = INF;
}
}
int u,v,w;
for(int i = 0; i < m; i++)
{
scanf("%d %d %d",&u,&v,&w);
if(dis[u][v] > w) //注意重边
dis[u][v] = dis[v][u] = w;
}
int x;
sump = 0;
for(int i = 1; i <= n; i++)
{
scanf("%d",&x);
weight[i] = x; //每个基地的能量(价值)
sump += x; //总价值
}
sump = (sump>>1) + 1;
floyd();
sumv = 0;
for(int i = 1; i <= n; i++)
{
cost[i] = dis[0][i];//每个基地的代价(花费)
if(cost[i] != INF)
sumv += cost[i];//总代价(花费)
}
ZeroOnePack();
int f = 0;
for(int i = 1; i <= sumv; i++)
{
if(dp[i] >= sump)
{
printf("%d\n",i);
f = 1;
break;
}
}
if(!f) printf("impossible\n");
}
return 0;
}