hdu 3339 In Action(最短路径+01背包)

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;
}