PAT 1068 万绿丛中一点红

https://pintia.cn/problem-sets/994805260223102976/problems/994805265579229184

 

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

输入格式：

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

代码：

#include <bits/stdc++.h>
using namespace std;

int x[11] = {0, 1, 1, 1, 0, -1, -1, -1};
int y[11] = {1, 1, 0, -1, -1, -1, 0, 1};
int col[1111][1111];
int num[1 << 24] = {0};

int main() {
    int N, M, TOL;
    scanf("%d%d%d", &N, &M, &TOL);

    for(int i = 1; i <= M; i ++){
        for(int j = 1; j <= N; j ++) {
            scanf("%d", &col[i][j]);
            num[col[i][j]] ++;
        }
    }

    int cnt, temp1, temp2, ans;
    bool flag = true;
    cnt = 0;
    for(int i = 1; i <= M; i ++) {
        for(int j = 1; j <= N; j ++) {
            if(num[col[i][j]] != 1)
                continue ;
            else {
                flag = false;
                for(int k = 0; k <= 7; k ++) {
                    if(abs(col[i][j] - col[i + x[k]][j + y[k]]) <= TOL) {
                        flag = true; break;
                    }
                }
                if(!flag) {
                    cnt++;
                    temp1 = i; temp2 = j;
                }
            }
        }
    }

    //cout<< cnt <<endl;
    if(cnt > 1) printf("Not Unique\n");
    else if(cnt == 0) printf("Not Exist\n");
    else {
        printf("(%d, %d): %d\n", temp2, temp1, col[temp1][temp2]);
    }

    return 0;
}