题意:
给定n个点 m条无向边 d元。
下面m行表示每条边 u<=>v 以及花费 w
下面top
下面top行
num c d 表示点标为num的城市 工资为c 健康证价格为d
目标是经过给定的top个城市,当到达该城市时,必须马上购买该城市的健康证并打工赚钱(每个城市只打工1次)
问从1城市出发,最后回到1城市,能否收集到所有的健康证
思路:
由于top很小,所以状压dp
dp[i][tmp]表示当前处于i点 经过城市的状态为tmp时 身上最多的钱。
首先对dis数组floyd 跑出最短路,然后裸裸地转移。
加了堆优化才过。。。
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
#define ll int
#define inf 100000000
#define N 101
int dp[15][1<<15];
int go[N];
int n, m, d;
int dis[N][N];
int C[N], D[N];
int Stack[N], top;
void floyd(){
for(int z = 1; z <= n; z++)dis[z][z] = 0;
for(int k = 1; k <= n; k++)
for(int i = 1; i <= n; i++)if(dis[i][k] !=inf && i!=k)
for(int j = 1; j <= n; j++)if(dis[k][j]!=inf && j!=i && j!=k)
dis[i][j] = min(dis[i][j], dis[i][k]+dis[k][j]);
}
void init(){
memset(dp, -1, sizeof dp);
for(int i = 1; i <= n; i++)for(int j = 1; j <= n; j++)dis[i][j] = inf;
}
struct node{
int u, t, mon;
node(int a=0,int b=0,int c=0):u(a),t(b),mon(c){}
bool operator<(const node&a)const{
return a.mon>mon;
}
};
void BFS(){
priority_queue<node>q;
for(int i = 0; i < top; i++) if(d - dis[1][Stack[i]] - D[i]>=0)
q.push(node(Stack[i], go[Stack[i]], d-dis[1][Stack[i]]-D[i]+C[i]));
while(!q.empty())
{
node a = q.top(); q.pop();
for(int i = 0; i < top; i++){
int v = Stack[i];
if(a.t & go[v])continue;
node now = a; now.t |= go[v];
if(dp[i][now.t]==-1 && now.mon - dis[a.u][v] - D[i] >= 0)
{
now.mon = now.mon - dis[a.u][v] - D[i] + C[i];
dp[i][now.t] = now.mon;
q.push(now);
}
}
}
}
int main()
{
int i, T, j, u, v, dd;scanf("%d",&T);
while(T--){
scanf("%d %d %d",&n,&m,&d);
init();
while(m--){
scanf("%d %d %d",&u,&v,&dd);
dis[u][v] = dis[v][u] = min(dis[u][v],dd);
}
floyd();
scanf("%d",&top);
for(i=0;i<top;i++){
scanf("%d %d %d",&Stack[i],&C[i],&D[i]);
go[Stack[i]] = 1<<i;
}
BFS();
int ans = -1;
for(i = 0; i < top; i++){
v = dp[i][(1<<top)-1];
ans = max(v - dis[1][Stack[i]], ans);
}
ans>=0?puts("YES"):puts("NO");
}
return 0;
}
/*
2 1 1
1 2 1
2
1 100 1
2 1 0
*/HDU 4284 状压dp+spfa堆优化,布布扣,bubuko.com
原文地址:http://blog.csdn.net/acmmmm/article/details/25431943