城市統計【BFS】

題目大意：

中山市的地圖是一個n*n的矩陣，其中標號為1的表示商業區，標號為0的表示居民區。為了考察市內居民區與商業區的距離，并對此作出評估，市長希望你能夠編寫一個程序完成這一任務。
　　居民區i到商業區的距離指的是到距離它最近的商業區j的距離(|Xi-Xj|+|Yi-Yj|)，而你將統計的是對于城市中的每一個區域k，以它為中心，所有滿足max(|Xk-Xm|,|Yk-Ym|)<=r的區域m到商業區距離之和。結果同樣以n*n的矩陣形式輸出。

思路：

70分：?
O(n⁴)直接暴力求解即可。

100分：

BFS

首先在讀入的時候，若這個點是商業區，就先將它入隊，之后跑一邊BFS，求出每個居民區到商業區的距離（由于每個點只要訪問1次，所以時間復雜度為O(n²)），之后用二維前綴和加速，輸出每個位置的答案。總時間復雜度為O(tn²)。

代碼：

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;

const int dx[]={0,0,0,1,-1};
const int dy[]={0,1,-1,0,0};
int a[301][301],b[301][301],p[301][301],t,n,r,sum,head,tail,state[500001][3];

int abs(int x)
{
    if (x>0) return x;
    return -x;
}

int minn(int x)
{
    return min(x,n);
}

int maxn(int x)
{
    return max(x,0);
}

void bfs()
{
    do
    {
        head++; 
        for (int i=1;i<=4;i++)  //向四個方向擴展
        {
            int xx=state[head][1]+dx[i];
            int yy=state[head][2]+dy[i];
            if (xx<0||xx>n||yy<0||yy>n||p[xx][yy]) continue;
            tail++;  //入隊
            p[xx][yy]=1;
            state[tail][1]=xx;
            state[tail][2]=yy;
            b[xx][yy]=b[state[head][1]][state[head][2]]+1;
        }
    }
    while (head<tail);
    return;
}

int main()
{
    scanf("%d",&t);
    while (t--)
    {
        memset(b,0,sizeof(b));
        memset(a,0,sizeof(a));
        memset(p,0,sizeof(p));
        scanf("%d%d",&n,&r);
        for (int i=1;i<=n;i++)
         for (int j=1;j<=n;j++)
         {
            scanf("%d",&a[i][j]);
            a[i][j]++;
            if (a[i][j]==2)  //商業區
            {
                tail++;  //入隊
                state[tail][1]=i;
                state[tail][2]=j;
                p[i][j]=1;
            }
        } 
        bfs();
        for (int i=1;i<=n;i++)
         for (int j=1;j<=n;j++)
          b[i][j]+=b[i-1][j]+b[i][j-1]-b[i-1][j-1];  //前綴和
        for (int i=1;i<=n;i++)
        {
            for (int j=1;j<=n;j++)
             printf("%d ",b[minn(i+r)][minn(j+r)]-b[maxn(i-r-1)][minn(j+r)]-b[minn(i+r)][maxn(j-r-1)]+b[maxn(i-r-1)][maxn(j-r-1)]); 
            putchar(10);
        } 
        putchar(10);
    }
    return 0;
}

?