JZOJ 2137. 【GDKOI2004】城市统计 (Standard IO)

2137. 【GDKOI2004】城市统计 (Standard IO)

Time Limits: 1000 ms  Memory Limits: 128000 KB  Detailed Limits  

Goto ProblemSet

Description

　　中山市的地图是一个n*n的矩阵，其中标号为1的表示商业区，标号为0的表示居民区。为了考察市内居民区与商业区的距离，并对此作出评估，市长希望你能够编写一个程序完成这一任务。
　　居民区i到商业区的距离指的是到距离它最近的商业区j的距离(|Xi-Xj|+|Yi-Yj|)，而你将统计的是对于城市中的每一个区域k，以它为中心，所有满足max(|Xk-Xm|,|Yk-Ym|)<=r的区域m到商业区距离之和。结果同样以n*n的矩阵形式输出。

 

Input

         第一行为t，表示以下有t组数据，每组数据之间以空行隔开，以下：
　　　第一行为n,r(1<=r<n<=150) 　　
          第二行起为一个n*n的矩阵。

Output

　　t组n*n的矩阵。每组用空行隔开

 

Sample Input

Sample Input1：
1
4 1
1 0 0 0
1 1 0 0
0 1 1 0
0 1 0 0

Sample Input2：
2
10 4
1 0 0 0 0 0 0 0 0 1
0 1 0 0 0 0 0 0 1 0
0 0 1 0 0 0 0 1 0 0
0 0 0 1 0 0 1 0 0 0
0 0 0 0 1 1 0 0 0 0
0 0 0 0 1 1 0 0 0 0
0 0 0 1 0 0 1 0 0 0
0 0 1 0 0 0 0 1 0 0
0 1 0 0 0 0 0 0 1 0
1 0 0 0 0 0 0 0 0 1

10 9
1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1
1 1 1 1 1 1 1 1 1 1

Sample Output

Sample Output1：
1 4 9 8
2 5 10 9
2 4 7 7
2 3 4 4

Sample Output2：
40 50 57 63 70 70 63 57 50 40 
50 60 68 76 86 86 76 68 60 50 
57 68 76 85 97 97 85 76 68 57 
63 76 85 94 107 107 94 85 76 63 
70 86 97 107 120 120 107 97 86 70 
70 86 97 107 120 120 107 97 86 70 
63 76 85 94 107 107 94 85 76 63 
57 68 76 85 97 97 85 76 68 57 
50 60 68 76 86 86 76 68 60 50 
40 50 57 63 70 70 63 57 50 40 

0 0 0 0 0 0 0 0 0 0 
0 0 0 0 0 0 0 0 0 0 
0 0 0 0 0 0 0 0 0 0 
0 0 0 0 0 0 0 0 0 0 
0 0 0 0 0 0 0 0 0 0 
0 0 0 0 0 0 0 0 0 0 
0 0 0 0 0 0 0 0 0 0 
0 0 0 0 0 0 0 0 0 0 
0 0 0 0 0 0 0 0 0 0 
0 0 0 0 0 0 0 0 0 0 

 

做法：先用bfs求出每个点的权值，然后跑一遍二位前缀和就好啦。

 

代码如下：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdlib>
#include <algorithm>
#define N 157
using namespace std;
int dis[N][N], dist[N][N], n, r, T, list[N * N][2], map[N][N];
bool v[N][N];
int dx[4] = {0, 1, 0, -1};
int dy[4] = {1, 0, -1, 0};

inline int max(int a, int b) {    return a > b ? a : b; }
inline int min(int a, int b) {    return a < b ? a : b; }

void readln()
{
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= n; j++)
                scanf("%d", &map[i][j]);    
}

inline bool    check(int x, int y)
{
    if (x >= 1 && y >= 1 && x <= n && y <= n && !v[x][y])    return 1;
    return 0;
}

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

inline int bfs(int x, int y)
{
    int head = 0, tail = 0;
    list[++tail][0] = x, list[tail][1] = y;
    while (head <= tail)
    {
        head++;
        int q = list[head][0], p = list[head][1];
        for (int i = 0; i <= 3; i++)
            if (check(q + dx[i], p + dy[i]))
            {
                v[q + dx[i]][p + dy[i]] = 1;
                if (map[q + dx[i]][p + dy[i]])
                {
                    for (int j = 1; j <= tail; j++)
                        v[list[j][0]][list[j][1]] = 0;
                    v[q + dx[i]][p + dy[i]] = 0;
                    return abs(q + dx[i] - x) + abs(p + dy[i] - y);
                }
                list[++tail][0] = q + dx[i];
                list[tail][1] = p + dy[i];
            }
    }
}

void xgm()
{
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= n; j++)
            dist[i][j] = dist[i - 1][j] + dist[i][j - 1] - dist[i - 1][j - 1] + dis[i][j]; 
}

void writeln()
{
    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= n - 1; j++)
            printf("%d ", dist[min(i + r, n)][min(j + r, n)] - dist[min(i + r, n)][max(j - r - 1, 0)] - dist[max(i - r - 1, 0)][min(j + r, n)] + dist[max(i - r - 1, 0)][max(j - r - 1, 0)]); 
        printf("%d
", dist[min(i + r, n)][min(n + r, n)] - dist[min(i + r, n)][max(n - r - 1, 0)] - dist[max(i - r - 1, 0)][min(n + r, n)] + dist[max(i - r - 1, 0)][max(n - r - 1, 0)]);
    }            
}

void work()
{
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= n; j++)
            if (!map[i][j])    dis[i][j] = bfs(i, j);
    xgm();
    writeln();
}

int main()
{
    scanf("%d", &T);
    while (T--)
    {
        memset(dis, 0, sizeof(dis));
        memset(dist, 0, sizeof(dist));
        memset(map, 0, sizeof(map));
        scanf("%d%d", &n, &r);
        readln();
        work();    
        if (T != 0)    printf("
");
    }
}