hdu 3339 In Action

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=3339

解题思路

1）floyd求最短路径--基地0到其它n个电厂的最短路径

2）01背包-----把路径的总和当成容量总和，把电量价值，求最少容量的最大价值

经典的 dp[j]=min(dp[j],dp[j-p[i]]+dis[i])

#include<stdio.h>
#include<string.h>
#define N 20000000
int map[105][105];
int p[105];
int dis[105];
int dp[10005];
int min(int a,int b)
{
	if(a<b)
		return a;
	else
		return b;
}
int max(int a,int b)
{
	if(a>b)
	return a;
	else
	return b;

}

int main()
{
    int t,n,m,i,j,k,a,b,w,sum,mi;
    scanf("%d",&t);
    while(t--)
	{

         scanf("%d%d",&n,&m);
         memset(map,0,sizeof(map));
         for(i=0;i<=n;i++)
          for(j=0;j<=n;j++)
           map[i][j]=N;
         for(i=0;i<m;i++)
		 {
            scanf("%d%d%d",&a,&b,&w);
           if(map[a][b]>w)
		   {
            map[a][b]=w;
            map[b][a]=w;
		   }
		 }
         sum = 0;
        for(i=1;i<=n;i++)
		{
           scanf("%d",&p[i]);
           sum+=p[i];
		}
        sum=sum/2+1;
        for(i=0;i<=n;i++)
           map[i][i]=0;
        for(k=0;k<=n;k++)
          for(i=0;i<=n;i++)
             for(j=0;j<=n;j++)
           map[i][j]=min(map[i][j],map[i][k]+map[k][j]);

       for(i=1;i<=n;i++)
           dis[i]=map[0][i];
          for(i=0;i<=10005;i++)
		  {
            dp[i]=N;
		  }
          dp[0]=0;
         for(i=1;i<=n;i++)
		 {
          if(dis[i]!=N)
		  {
            for(j=10000;j>=p[i];j--)
			{
            dp[j]=min(dp[j],dp[j-p[i]]+dis[i]);
			}
		  }
		 }
          mi=N;
         for(i=sum;i<=10000;i++)
		 {
               if(dp[i]!=N)
			   {
                 if(dp[i]<mi)
                   mi=dp[i];
			   }
		}
         if(mi!=N)
          printf("%d\n",mi);
          else
        printf("impossible\n");

	}
 return 0;
}