UVALive 5066 Fire Drill --BFS+DP
题意：有一个三维的地图，有n个人被困住，现在消防队员只能从1楼的一个入口进入，营救被困者，每一个被困者有一个价值，当消防队员找到一个被困者之后，他可以营救或者见死不救，如果救的话，他必须马上将其背到入口处，不得停下，不得同时救多个人，而且回去的时间一步要做两步走，即时间增加一倍。求在给定时间S内，能救到的人的最大价值总和。
解法：bfs一遍记录每个点离起点的最短距离，那么救这个人的花费就是3*dis，然后已经知道救这个人的价值，那么最后求一个01背包即可。
要注意一个人都救不到的地方，我开始将dis都初始化为1000000007,这样的话如果走不到的话，花费就变成3*dis爆int了。悲剧。
代码：
#include <iostream> #include <cstdio> #include <cstring> #include <cstdlib> #include <cmath> #include <algorithm> #include <queue> #define Mod 10000007 using namespace std; #define N 50107 char mp[15][125][125]; int dis[15][125][125]; int dp[10006]; int T[114],P[114],L,H,W; int dx[7] = {0,0,0,0, 1, -1}; int dy[7] = {0,1,-1,0, 0, 0,}; int dz[7] = {1,0,0,-1, 0, 0,}; struct node { int l,h,w,sec; node(int _x,int _y,int _z,int _sec):l(_x),h(_y),w(_z),sec(_sec){} node(){} }vol[135],S; bool OK(int x,int y,int z) { if(x >= 0 && x < L && y >= 0 && y < H && z >= 0 && z < W) return true; return false; } void bfs(node S) { queue<node> q; while(!q.empty()) q.pop(); int i,j,k; for(i=0;i<L;i++) for(j=0;j<H;j++) for(k=0;k<W;k++) dis[i][j][k] = Mod; int x = S.l, y = S.h, z = S.w; dis[x][y][z] = 0; q.push(S); while(!q.empty()) { node now = q.front(); q.pop(); int l = now.l, h = now.h, w = now.w, sec = now.sec; if(mp[l][h][w] == 'U') { for(k=0;k<5;k++) { int nx = l + dx[k]; int ny = h + dy[k]; int nz = w + dz[k]; if(!OK(nx,ny,nz)) continue; if(mp[nx][ny][nz] != 'X') { if(sec+1 < dis[nx][ny][nz]) { dis[nx][ny][nz] = sec+1; node tmp = node(nx,ny,nz,sec+1); q.push(tmp); } } } } else if(mp[l][h][w] == 'D') { for(k=0;k<6;k++) { if(k == 4) continue; int nx = l + dx[k]; int ny = h + dy[k]; int nz = w + dz[k]; if(!OK(nx,ny,nz)) continue; if(mp[nx][ny][nz] != 'X') { if(sec+1 < dis[nx][ny][nz]) { dis[nx][ny][nz] = sec+1; node tmp = node(nx,ny,nz,sec+1); q.push(tmp); } } } } else { for(k=0;k<4;k++) { int nx = l + dx[k]; int ny = h + dy[k]; int nz = w + dz[k]; if(!OK(nx,ny,nz)) continue; if(mp[nx][ny][nz] != 'X') { if(sec+1 < dis[nx][ny][nz]) { dis[nx][ny][nz] = sec+1; node tmp = node(nx,ny,nz,sec+1); q.push(tmp); } } } } } } int main() { int t,n,STime,i,j,k; scanf("%d",&t); while(t--) { scanf("%d%d%d%d%d",&L,&H,&W,&n,&STime); for(i=0;i<L;i++) { for(j=0;j<H;j++) { scanf("%s",mp[i][j]); for(k=0;k<W;k++) if(mp[i][j][k] == 'S') S = node(i,j,k,0); } } for(i=1;i<=n;i++) { scanf("%d%d%d%d",&vol[i].l,&vol[i].h,&vol[i].w,&P[i]); vol[i].l--,vol[i].h--,vol[i].w--; } bfs(S); for(i=1;i<=n;i++) T[i] = 3*dis[vol[i].l][vol[i].h][vol[i].w]; memset(dp,0,sizeof(dp)); for(i=1;i<=n;i++) { for(j=STime;j>=0;j--) { if(j >= T[i]) dp[j] = max(dp[j],dp[j-T[i]]+P[i]); } } int Maxi = 0; for(i=0;i<=STime;i++) Maxi = max(Maxi,dp[i]); printf("%d\n",Maxi); } return 0; }