【Lintcode】390. Find Peak Element II

题目描述

法1：
对行二分。我们选取A AA的中间一行，并寻找该行的最大值，如果其为峰值，则直接返回其坐标；否则的话要么其上方比其更大，要么其下方比其更大。如果其上方比其更大，那么A AA的上半部分就一定有一个峰值（证明很简单，顺着这个值，每次走向与其相邻的最大值，由于上半部分数字只有有限个，这样走有限步一定会停下来，并且一定不会停在边界上，也不会走到下半部分，因为我们是从中间行的最大值出发的。停下来的地方就是峰值）。这样我们就将问题规模缩小成为了一半。如此缩小下去即能找到峰值坐标。代码如下：

import java.util.ArrayList;
import java.util.List;

public class Solution {
    /*
     * @param A: An integer matrix
     * @return: The index of the peak
     */
    public List<Integer> findPeakII(int[][] A) {
        // write your code here
        List<Integer> res = new ArrayList<>();
        
        int l = 1, r = A.length - 2;
        while (l <= r) {
            int m = l + (r - l >> 1);
            // 找中间行的最大值和其坐标
            int max = Integer.MIN_VALUE, maxIdx = 0;
            for (int i = 0; i < A[m].length; i++) {
                if (A[m][i] > max) {
                    max = A[m][i];
                    maxIdx = i;
                }
            }
            
            // 由于峰值一定存在，所以最后一定会从这里break出来
            if (max > A[m - 1][maxIdx] && max > A[m + 1][maxIdx]) {
                res.add(m);
                res.add(maxIdx);
                break;
            }
            
            if (max < A[m - 1][maxIdx]) {
                r = m - 1;
            } else {
                l = m + 1;
            }
        }
        
        return res;
    }
}

时间复杂度O ( n log ⁡ m ) O(n\log m)O(nlogm)，空间O ( 1 ) O(1)O(1)。
法2：四分法。选中中间的行和中间的列，找到这个“十字架”的最大值。如果这个最大值就是峰值则输出其坐标，否则看一下其上下或是左右哪个数比它大，就去到那个数所在的象限去找峰值。代码如下：

import java.util.ArrayList;
import java.util.List;

public class Solution {
    /*
     * @param A: An integer matrix
     * @return: The index of the peak
     */
    public List<Integer> findPeakII(int[][] A) {
        // write your code here
        List<Integer> res = new ArrayList<>();
        
        int u = 1, d = A.length - 2, l = 1, r = A[0].length - 2;
        
        while (u <= d && l <= r) {
        	// 取出中间行和中间列
            int mRow = u + (d - u >> 1), mCol = l + (r - l >> 1);
            
            // 找十字架的最大值和其所在坐标
            int max = Integer.MIN_VALUE, maxRow = 0, maxCol = 0;
            for (int i = l; i <= r; i++) {
                if (A[mRow][i] > max) {
                    max = A[mRow][i];
                    maxRow = mRow;
                    maxCol = i;
                }
            }
            for (int i = u; i <= d; i++) {
                if (A[i][mCol] > max) {
                    max = A[i][mCol];
                    maxRow = i;
                    maxCol = mCol;
                }
            }
            
            // 如果是峰值，则返回答案
            if (max > A[maxRow - 1][maxCol] && max > A[maxRow + 1][maxCol] && max > A[maxRow][maxCol - 1] && max > A[maxRow][maxCol + 1]) {
                res.add(maxRow);
                res.add(maxCol);
                break;
            }
            
            // 如果在十字架的行上，则上下做比较；否则左右做比较
            if (maxRow == mRow) {
            	// 先确定在上下哪半
                if (A[mRow - 1][maxCol] > max) {
                    d = mRow - 1;
                } else if (A[mRow + 1][maxCol] > max) {
                    u = mRow + 1;
                }
                // 再确定是左右哪半
                if (maxCol < mCol) {
                    r = maxCol;
                } else {
                    l = maxCol;
                }
            } else if (maxCol == mCol) {
                if (A[maxRow][mCol - 1] > max) {
                    r = mCol - 1;
                } else if (A[maxRow][mCol + 1] > max) {
                    l = mCol + 1;
                }
                
                if (maxRow < mRow) {
                    d = maxRow;
                } else {
                    u = maxRow;
                }
            }
        }
        
        return res;
    }
}

时间复杂度O ( m + n ) O(m+n)O(m+n)，可以由以下式子得出：

空间复杂度O(1)。