CodeForces - 1105D 多源搜索

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
struct node{
    int x,y;
    ll setp;
};
queue<node> Q,border[];
const int maxn=;
char a[maxn][maxn];
int vis[maxn][maxn];
int n,m,p;
ll ans[],s[];
int f[][]={
        {,},{,},{,-},{-,}
};
void bfs(int w)
{
    node temp;
    int x,y;
    while(!Q.empty())
    {
        temp=Q.front();
        Q.pop();
        if(temp.setp==)
            border[w].push(temp);
        else{
            for(int i=;i<;i++)
            {
                x=temp.x+f[i][];
                y=temp.y+f[i][];
                if(x<||x>n||y<||y>m||vis[x][y]!=||a[x][y]!='.') continue;
                vis[x][y]=w;
                Q.push(node{x,y,temp.setp-});
            }
        }
    }
}
bool expand(int w)
{
    node temp;
    while(!border[w].empty())
    {
        temp=border[w].front();
        border[w].pop();
        temp.setp=s[w];
        Q.push(temp);
    }
    bfs(w);
    return !border[w].empty();
}
int main()
{
    cin>>n>>m>>p;
    for(int i=;i<=p;i++)
        cin>>s[i];
    for(int i=;i<=n;i++)
        cin>>a[i]+;
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=m;j++)
        {
            if(a[i][j]>=''&&a[i][j]<='')
            {
                int num=a[i][j]-'';
                border[num].push(node{i,j,s[num]});
                vis[i][j]=num;
            }
        }
    }
    while()
    {
        bool flag= false;
        for(int i=;i<=p;i++)
            flag|=expand(i);
        if(!flag)
            break;
    }
    for(int i=;i<=n;i++)
        for(int j=;j<=m;j++)
            ans[vis[i][j]]++;
    for(int i=;i<=p;i++)
        cout<<ans[i]<<" ";
}