Codeforces 1105D(Kilani and the Game,双队列bfs)

#include<bits/stdc++.h>
#define ll long long
#define endl ‘\n‘
#define mem(a,b) memset(a,b,sizeof(a))
#define IO ios::sync_with_stdio(false);cin.tie(0);
using namespace std;
const int INF=0x3f3f3f3f;
const ll inf=0x3f3f3f3f3f3f3f3f;
const int mod=1e9+7;
const int maxn=1e5+5;
struct node{
    int x,y,t;
    node(int xx,int yy,int tt){
        x=xx;y=yy;t=tt;
    }
};
int s[15],vis[1005][1005];char a[1005][1005];
int n,m,k;
queue<node> q1[15];
queue<node> q2[15];
int dx[4]={1,0,0,-1};
int dy[4]={0,-1,1,0};
int bfs(int p){
    int newx=0;
    while(!q2[p].empty()){
        node x=q2[p].front();q2[p].pop();
        x.t=0;
        q1[p].push(x);
    }
    while(!q1[p].empty()){
        node x=q1[p].front();q1[p].pop();
        if(x.t==s[p]){
            q2[p].push(x);
            continue;
        }
        for(int i=0;i<4;i++){
            int xx=x.x+dx[i];
            int yy=x.y+dy[i];
            if(xx<1||xx>n||yy<1||yy>m||a[xx][yy]==‘#‘||vis[xx][yy]||x.t+1>s[p]) continue;
            newx+=1;
            q1[p].push(node(xx,yy,x.t+1));
            vis[xx][yy]=p;
        }
    }
    if(newx>=1) return 1;
    else return 0;    
}
int ans[15];
int main(){
    cin>>n>>m>>k;
    for(int i=1;i<=k;i++) cin>>s[i];
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            cin>>a[i][j];
            if(a[i][j]-‘0‘>=1&&a[i][j]-‘0‘<=9){
                vis[i][j]=a[i][j]-‘0‘;
                q2[a[i][j]-‘0‘].push(node(i,j,0));
            }
        }
    }
    while(1){
        int flag=0;
        for(int i=1;i<=9;i++){
            flag+=bfs(i);
        }
        if(flag==0) break;
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            ans[vis[i][j]]+=1;
        }
    }
    for(int i=1;i<=k;i++){
        cout<<ans[i]<<" ";
    }
    cout<<endl;
}