PAT (Basic Level) Practice 1068

1068 万绿丛中一点红（20 分）

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

输入格式：

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

分析：用二维数组储存输入数据，用map记录每个值出现的次数。因为所需点的个数必须是1，所有值不为1的点都不用考虑。用一个8行2列的数组表示8个方向，在搜索的时候直接套方向。因为数据量大，合法性判断最好放在套方向的循环里面，否则容易超时。

代码：

#include<iostream>
#include<cmath>
#include<map>
#define MAXN 1001
using namespace std;
int M, N, TOL;
int x, y, value;
int arr[MAXN][MAXN] = { 0 };
int dire[8][2] = { {1, 0}, {-1, 0}, {0, 1}, {0, -1}, {1, 1}, {1, -1}, {-1, 1}, {-1, -1} };
map<int, int> m;
void search() {
	bool isExist = false;
	bool isUnique = false;
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			if (m[arr[i][j]] == 1) {
					bool isVaild = true;
					for (int k = 0; k < 8; k++) {
						int ni = i + dire[k][0];
						int nj = j + dire[k][1];
						if (ni >= 0 && nj >= 0 && ni < N && nj < M && abs(arr[ni][nj] - arr[i][j]) <= TOL) {
							isVaild = false;
							break;
						}
					}
					if (isVaild == true && isUnique == false) {
						x = j + 1;
						y = i + 1;
						value = arr[i][j];
						isUnique = true;
						isExist = true;
					}else if (isVaild == true && isUnique == true) {
						cout << "Not Unique" << endl;
						return;
					}
			}
		}
	}
	if (isExist == false) {
		cout << "Not Exist" << endl;
	}else {
		printf("(%d, %d): %d\n", x, y, value);
	}
}
int main() {
	cin >> M >> N >> TOL;
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			cin >> arr[i][j];
			m[arr[i][j]]++;
		}
	}
	search();
	return 0;
}