题目链接:
http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=2849
题解:
题目大意:有一个N*M大小的矩阵,里面有一些电脑,当某个点的权值ai大于0的时,表示这个电脑已经中了ai的这个病毒,如果ai是负数,则表示这个电脑还没有中病毒,并且它的绝对值就是它的防护的等级。然后会有许多天,每一天,病毒只能转给防护等级小于等于天数的相邻的电脑,并且每一次,病毒的转播都是从编号较小的病毒开始的。问,当所有的电脑都已经中了病毒的时候,不同的病毒感染了多少电脑。
本题的难点在于后面的BFS的剪枝,不然很容易会出现超时的现象,
这里我就举一个例子:如果一个病毒的四周有三个防御等级是-1的和一个防御等级是-100的,那么如果我们还是以每天都+1的方式去判断,那么到100的时候会很浪费时间。所以,我们这里直接记录一个最小值,然后去直接更新最小值。
代码:
#include <bits/stdc++.h>
using namespace std;
#define met(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
const int maxn = 500+10;
int mp[maxn][maxn];
int visited[maxn][maxn];
int num[500005];
int dir[4][2]={1,0,-1,0,0,1,0,-1};
struct node
{
int x,y;
int hurt;
int kind;
bool friend operator<(node a,node b) {
{
if(a.hurt!=b.hurt)
return a.hurt>b.hurt;
else
return a.kind>b.kind;
}
}
};
int n,m;
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
priority_queue<node> q;
while(!q.empty())
{
q.pop();
}
met(visited,0);
met(num,0);
int ans=0;
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
{
scanf("%d",&mp[i][j]);
if(mp[i][j]>0)
{
ans++;
node step;
step.x=i;
step.y=j;
step.hurt=1;
visited[i][j]=1;
step.kind=mp[i][j];
q.push(step);
continue;
}
}
}
//printf("%d\n",q.size());
/*while(!q.empty())
{
node s=q.top();
printf("%d\n",s.kind);
q.pop();
}*/
int cnt=0;
node now,next;
while(!q.empty())
{
now=q.top();
q.pop();
int MIN=-99999999;
for(int i=0;i<4;i++)
{
next.x=now.x+dir[i][0];
next.y=now.y+dir[i][1];
//next.hurt=now.hurt;
if(next.x>=0&&next.x<n&&next.y>=0&&next.y<m&&!visited[next.x][next.y]&&abs(mp[next.x][next.y])<=now.hurt)
{
visited[next.x][next.y]=1;
next.hurt=now.hurt;
//printf("%d %d %d\n",next.x,next.y,next.hurt);
cnt++;
next.kind=now.kind;
num[now.kind]++;
q.push(next);
}
else
{
if(MIN<mp[next.x][next.y]&&next.x>=0&&next.x<n&&next.y>=0&&next.y<m&&!visited[next.x][next.y])
{
MIN=mp[next.x][next.y];
}
}
}
if(MIN!=-99999999)
{
now.hurt=MIN*(-1);
//now.hurt+=1;
q.push(now);
}
if(cnt>=n*m-ans)
{
break;
}
}
int k;
scanf("%d",&k);
while(k--)
{
int x;
scanf("%d",&x);
printf("%d\n",num[x]+=1);
}
}
}
/*
3 4
1 -3 -2 -3
-2 -1 -2 2
-3 -2 -1 -1
2
1
2
*/