PAT 1068. 万绿丛中一点红(20)

对于计算机而言，颜色不过是像素点对应的一个24位的数值。现给定一幅分辨率为MxN的画，要求你找出万绿丛中的一点红，即有独一无二颜色的那个像素点，并且该点的颜色与其周围8个相邻像素的颜色差充分大。

输入格式：

输入第一行给出三个正整数，分别是M和N（<= 1000），即图像的分辨率；以及TOL，是所求像素点与相邻点的颜色差阈值，色差超过TOL的点才被考虑。随后N行，每行给出M个像素的颜色值，范围在[0, 224)内。所有同行数字间用空格或TAB分开。

输出格式：

在一行中按照“(x, y): color”的格式输出所求像素点的位置以及颜色值，其中位置x和y分别是该像素在图像矩阵中的列、行编号（从1开始编号）。如果这样的点不唯一，则输出“Not Unique”；如果这样的点不存在，则输出“Not Exist”。

输入样例1：

8 6 200
0 0 0 0 0 0 0 0
65280 65280 65280 16711479 65280 65280 65280 65280
16711479 65280 65280 65280 16711680 65280 65280 65280
65280 65280 65280 65280 65280 65280 165280 165280
65280 65280 16777015 65280 65280 165280 65480 165280
16777215 16777215 16777215 16777215 16777215 16777215 16777215 16777215

输出样例1：

(5, 3): 16711680

输入样例2：

4 5 2
0 0 0 0
0 0 3 0
0 0 0 0
0 5 0 0
0 0 0 0

输出样例2：

Not Unique

输入样例3：

3 3 5
1 2 3
3 4 5
5 6 7

输出样例3：

Not Exist

为了便于处理，将数组开大两个长度，这样就可以不用进行边界判定了，减少了代码的复杂度。isGreatThanTOL函数检测当前像素是否与周边8个像素点的差大于tol。在主函数中，用unique记录颜色出现的次数，我们只去判定那些独一无二的颜色，如果颜色既是独一无二的又是比周边的像素颜色差充分大的，那么这个点就是我们要考虑的了。

#include <cstdio>
#include <map>
#include <vector>
#include <cmath>

using namespace std;

vector<vector<int> > pic;

bool isGreatThanTOL(int a, int b, int tol) {
​    if (fabs(pic[a][b] - pic[a - 1][b])     <= tol) return false;
​    if (fabs(pic[a][b] - pic[a - 1][b + 1]) <= tol) return false;
​    if (fabs(pic[a][b] - pic[a - 1][b - 1]) <= tol) return false;
​    if (fabs(pic[a][b] - pic[a][b + 1])     <= tol) return false;
​    if (fabs(pic[a][b] - pic[a][b - 1])     <= tol) return false;
​    if (fabs(pic[a][b] - pic[a + 1][b + 1]) <= tol) return false;
​    if (fabs(pic[a][b] - pic[a + 1][b])     <= tol) return false;
​    if (fabs(pic[a][b] - pic[a + 1][b - 1]) <= tol) return false;
​    return true;
}

int main() {
​    int m = 0, n = 0, tol = 0;
​    scanf("%d %d %d", &m, &n, &tol);
​    
    pic.resize(n + 2);
    for (int i = 0; i < n + 2; i++) {
        pic[i].resize(m + 2);
    }
    
    map<int, int> unique;
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++) {
            scanf("%d", &pic[i][j]);
            unique[pic[i][j]]++;
        }
    }
    
    int a = 0, b = 0, cnt = 0;
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= m; j++) {
            if (unique[pic[i][j]] == 1 && isGreatThanTOL(i, j, tol)) {
                cnt++;
                a = i;
                b = j;
            }
        }
    }
    
    if (cnt == 0) {
        printf("Not Exist\n");
    } else if (cnt == 1) {
        printf("(%d, %d): %d\n", b, a, pic[a][b]);
    } else {
        printf("Not Unique\n");
    }
    return 0;
}