zoj 1849 (浙江省赛)Attack of Panda Virus

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=1849

/*思路 ：

   优先队列 bfs ，从每一天中 病毒类型最小的开始 ，所以优先队列 

   先 按天 排在按类型  

*/



#include<iostream>

#include<stdio.h>

#include<queue>

#include<cmath>

#include<string.h>

#define N 600

using namespace std;

struct node

{

    int x;

    int y;

    int day;

    int ty;

    friend bool  operator < (struct node a,struct node b)

    {

        if(a.day!=b.day)

         return  a.day>b.day;

         else

          return  a.ty>b.ty;

    }

};

int map[N][N],d[4][2]={{0,-1},{-1,0},{0,1},{1,0}};

int num[300000],n,m,sum;

priority_queue<node>q;

node p;

void init()

{

      int i,j;

    while(!q.empty())

       q.pop();



      sum=0;

    memset(num,0,sizeof(num));

     for(i=0;i<1500;i++)

       num[i]=0;



       for(i=0;i<n;i++)

       {



           for(j=0;j<m;j++)

           {

               scanf("%d",&map[i][j]);

               if(map[i][j]>0)

               {

                   p.x=i;

                   p.y=j;

                   p.day=1;

                   p.ty=map[i][j];

                   num[p.ty]++;

                   sum++;

                   q.push(p);



               }

           }

       }



}

bool bound(int x,int y)

{

    if(x>=0&&y>=0&&x<n&&y<m)

     return true;

     else  return false;

}

void bfs()

{

    node temp;

    int i;



    while(!q.empty())

    {

        p=q.top();

        q.pop();

        int max=-1111111;

      for(i=0;i<4;i++)

      {

          temp.x=p.x+d[i][0];

          temp.y=p.y+d[i][1];

          temp.day=p.day;

          temp.ty=p.ty;

          if(!bound(temp.x,temp.y))continue;



          if(map[temp.x][temp.y]<0)

          {

              if(p.day>=fabs(map[temp.x][temp.y]))

              {

                  map[temp.x][temp.y]=p.ty;

                  num[p.ty]++;

                  sum++;

                  //printf("%d\n",sum);

                  if(sum==m*n){return ;}

                  q.push(temp);



              }

              else

              {

                  if(map[temp.x][temp.y]>max)//寻找周为最容易感染的

                

                     max=map[temp.x][temp.y];

              }



          }





      }

      if(max!=-1111111)

      {



          p.day=-max;   //若不能感染周围的 ，则等下一天

          q.push(p);

      }



    }

}

int main()

{

    int t,b;

    while(cin>>n>>m)

    {

        init();

        bfs();

        cin>>t;

        while(t--)

        {

            scanf("%d",&b);

            printf("%d\n",num[b]);

        }

    }

}