POJ 2227 -- The Wedding Juicer（bfs+优先队列）

题目大意：一个W*H的网格，每个单位格的高度不一样，往这个网格注水，问能储存多少水；

思路分析：四周不能注水，和木桶原理一样，要以最低的高度作为能储水的高度，否则水就会溢出；将网格最外层的点开始加入队列，每次去高度最小的点作为“木桶”最低边，看其连接的点，如果高度大于自己加入队列，否则注水至自己的高度加入队列。

代码实现：

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
int w[310][310],vis[310][310],n,m;
int d[4][2]={1,0,-1,0,0,1,0,-1};
long long sum;
struct Node{
    int x,y,h;
    Node(int xx=0,int yy=0,int hh=0):x(xx),y(yy),h(hh){}
    bool operator<(const Node &a)const{
        return h>a.h;
    }
};
void bfs(){
    priority_queue<Node> q;
    for(int i=1;i<=n;i++){
        q.push(Node(1,i,w[1][i]));
        q.push(Node(m,i,w[m][i]));
        w[1][i]=-1;
        w[m][i]=-1;
    }
    for(int i=1;i<=m;i++){
        q.push(Node(i,1,w[i][1]));
        q.push(Node(i,n,w[i][n]));
        w[i][1]=-1;
        w[i][n]=-1;
    }
    Node head;
    while(!q.empty()){
        head=q.top();
        q.pop();
        //printf("h %d %d\n",head.x,head.y);
        for(int i=0;i<4;i++){
            int x=head.x+d[i][0];
            int y=head.y+d[i][1];
            if(x<=1||x>=m||y<=1||y>=n) continue;
            if(w[x][y]==-1) continue;
            if(w[x][y]<head.h){
                sum+=head.h-w[x][y];
                //printf("*%d %d , %d %d\n",x,y,head.x,head.y);
                q.push(Node(x,y,head.h));
            }
            else q.push(Node(x,y,w[x][y]));
            w[x][y]=-1;
        }
    }
    printf("%I64d\n",sum);
}
int main(){
    while(~scanf("%d%d",&n,&m)){
        memset(w,-1,sizeof(w));
        memset(vis,0,sizeof(vis));
        sum=0;
        for(int i=1;i<=m;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&w[i][j]);
        bfs();
    }
}