PAT乙级-1068 万绿丛中一点红 (20 分)

题目链接

#include<iostream>
#include<vector>
#include<map>
using namespace std;

int m,n,tol;
vector<vector<int>> v;
int row[8]={-1,-1,-1,0,1,1,1,0}; //增量数组,左上方开始顺时针遍历
int col[8]={-1,0,1,1,1,0,-1,-1};

bool judge(int x,int y){
    for(int k=0;k<8;k++){
        int dx=x+row[k];
        int dy=y+col[k]; //(dx,dy)在图像内部且与(x,y)点色差不够
        if(dx>=0&&dx<n&&dy>=0&&dy<m&&abs(v[dx][dy]-v[x][y])<=tol) return false;  
    }
    return true;
}

int main(){
    cin>>m>>n>>tol;
    v.resize(n,vector<int>(m)); //n行m列
    map<int,int> mp; //颜色需独一无二
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>v[i][j];
            mp[v[i][j]]++;
        }
    }
    int x,y,cnt=0;
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            if(mp[v[i][j]]==1&&judge(i,j)){
                x=i+1;
                y=j+1;
                cnt++;
            }
        }
    }
    if(cnt==0) cout<<"Not Exist"<<endl;
    else if(cnt==1) printf("(%d, %d): %d\n",y,x,v[x-1][y-1]); //注意是列、行
    else cout<<"Not Unique"<<endl;
    return 0;
}