HDU 3339 In Action

AC大神想要引爆核弹毁灭世界，可是在吹牛皮的时候不小心说漏了自己的计划，传着传着传到了奥巴马的耳朵里，尼玛，哥当上总统没几年还没享受完你就想毁灭世界？

AC大神有n个连通的发电站（编号为1~n），每个发电站有各自的发电量值，AC大神要启动核武，至少要消耗全部发电站的总电量的一半。

奥巴马知道了这一切后，带上头盔领着数量充足的坦克部队，集合于基地（编号为0），打算派出一些坦克占领（是占领，有一辆坦克留在发电站才叫占领）掉AC大神一半以上的发电站，让AC大神没办法成功实行自己的计划。每一辆坦克移动一单位的距离就消耗一单位的油，题目给出我们一些相关数据（具体看input），要求我们判断奥巴马的行动能否成功，如果奥巴马的计划能成功，则输出最小的耗油量，否则，奥巴马就一口血吐在屏幕上，立即身亡。

先用Dijkstra算法算出从0号到每个发电站的最短路径，然后就是一个01背包，从0号到每个发电站的距离的和作为背包容量，每个发电站所具有的电量为价值。为了减少空间消耗，用滚动数组优化了一下。。最后扫一下，得到完成任务的最少的耗油量。。

#include <stdio.h>
#include <string.h>
#include <queue>
#include <utility>
#define MAXN 100
using namespace std;

const int inf=0x3fffffff;
int w[MAXN+5][MAXN+5],cost[1000005];
int power[MAXN+5],dis[MAXN+5];
int n;

typedef pair<int,int> pii;

void Dijkstra(int st){
    int i,x;
    pii t;
    priority_queue<pii,vector<pii>,greater<pii> > q;
    for(i=1;i<=n;i++)
        dis[i]=inf;
    dis[st]=0;
    q.push(make_pair(dis[st],st));
    while(!q.empty()){
        t=q.top();
        q.pop();
        x=t.second;
        if(t.first!=dis[x])
            continue;
        for(i=1;i<=n;i++){
            if(dis[i]>dis[x]+w[x][i]){
                dis[i]=dis[x]+w[x][i];
                q.push(make_pair(dis[i],i));
            }
        }
    }
}

int Max(int a,int b){
	return a>b?a:b;
}

void scani(int &num){
	char ch;
	int flag=1;
	while(ch=getchar(),(ch>'9'||ch<'0')&&(ch!='-'));
	if(ch=='-')
		flag=-1,num=0;
	else
		num=ch-'0';
	while(ch=getchar(),ch<='9'&&ch>='0')
		num=num*10+ch-'0';
	num*=flag;
}

int main(){
	int T,m,st,ed,val,psum,i,j,dsum;
	scani(T);
	while(T--){
		scani(n),scani(m);
		for(i=0;i<=n;i++)
			for(j=0;j<=n;j++)
				w[i][j]=w[j][i]=inf;
		while(m--){
			scani(st),scani(ed),scani(val);
			if(w[st][ed]>val)
				w[st][ed]=w[ed][st]=val;
		}
		for(i=1;i<=n;i++)
			scani(power[i]);
		Dijkstra(0);
		if(dis[1]==inf){
			puts("impossible");
			continue;
		}
		dsum=psum=0;
		for(i=1;i<=n;i++){
			psum+=power[i];
			dsum+=dis[i];
		}
		memset(cost,0,(dsum+1)*sizeof(int));
		for(i=1;i<=n;i++)
			for(j=dsum;j>=dis[i];j--)
				cost[j]=Max(cost[j],cost[j-dis[i]]+power[i]);
		psum=(psum>>1)+1;
		for(i=1;i<=dsum;i++){
			if(cost[i]>=psum){
				printf("%d\n",i);
				break;
			}
		}
	}
	return 0;
}