1901. 寻找峰值 II-遍历矩阵

1901. 寻找峰值 II-遍历矩阵

一个 2D 网格中的 峰值 是指那些 严格大于 其相邻格子(上、下、左、右)的元素。

给你一个 从 0 开始编号 的 m x n 矩阵 mat ，其中任意两个相邻格子的值都 不相同 。找出 任意一个 峰值 mat[i][j] 并 返回其位置 [i,j] 。

你可以假设整个矩阵周边环绕着一圈值为 -1 的格子。

要求必须写出时间复杂度为 O(m log(n)) 或 O(n log(m)) 的算法

示例 1:

输入: mat = [[1,4],[3,2]]
输出: [0,1]
解释: 3 和 4 都是峰值，所以[1,0]和[0,1]都是可接受的答案。

示例 2:

输入: mat = [[10,20,15],[21,30,14],[7,16,32]]
输出: [1,1]
解释: 30 和 32 都是峰值，所以[1,1]和[2,2]都是可接受的答案。

这一题，其实就很简单了，我们遍历一下，然后判断一下就可以了，但是还有一个思路更不错的方法，我们可以直接找出矩阵的最大值，这样也可以解决问题，博主解题代码如下：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */



int* findPeakGrid(int** mat, int matSize, int* matColSize, int* returnSize){
     int *re=(int *)malloc(sizeof(int)*2);
     int direction[4][2]={{0,1},{-1,0},{1,0},{0,-1}};
     for(int i=0;i<matSize;i++){
         for(int j=0;j<matColSize[0];j++){
             int val=mat[i][j];
             int k;
             for( k=0;k<4;k++){
                 int x_next=i+direction[k][0];
                 int y_next=j+direction[k][1];
                 if(x_next>=0&&x_next<matSize&&y_next>=0&&y_next<matColSize[0]){
                     if(mat[x_next][y_next]>=val){
                         break;
                     }

                 }

             }
             if(k==4){
                   re[0]=i;
             re[1]=j;
             *returnSize=2;
             return re;

             }
           

         }
     }
     return re;

}