1. 題目

給定一個整數矩陣 A, 它有如下特性:

• 相鄰的整數不同
• 矩陣有 n 行 m 列。
• 對于所有的 i < n, 都有 A[i][0] < A[i][1] && A[i][m - 2] > A[i][m - 1]
• 對于所有的 j < m, 都有 A[0][j] < A[1][j] && A[n - 2][j] > A[n - 1][j]

我們定義一個位置 [i,j] 是峰值, 當且僅當它滿足:

• A[i][j] > A[i + 1][j] && A[i][j] > A[i - 1][j] && A[i][j] > A[i][j + 1] && A[i][j] > A[i][j - 1]

找到該矩陣的一個峰值元素, 返回它的坐標.

樣例 1: 輸入: [[1, 2, 3, 6, 5],[16,41,23,22, 6],[15,17,24,21, 7],[14,18,19,20,10],[13,14,11,10, 9]] 輸出: [1,1] 解釋: [2,2] 也是可以的. [1,1] 的元素是 41, 大于它四周的每一個元素 (2, 16, 23, 17).樣例 2: 輸入: [[1, 5, 3],[4,10, 9],[2, 8, 7]] 輸出: [1,1] 解釋: 只有這一個峰值挑戰 O(n+m) 的時間復雜度. 如果你 認為 你使用了 O(nlogm) 或 O(mlogn) 的算法, 能否證明它的復雜度其實是 O(n+m)? 或者想一個類似的算法但是復雜度是O(n+m)？注意事項 保證至少存在一個峰值, 而如果存在多個峰值, 返回任意一個即可.

2. 解題

class Solution { public:vector<int> findPeakII(vector<vector<int>> &A) {int n = A.size(), m = A[0].size();int i = 1, j = 1;//從(1,1)開始，（0,0）不可能是答案vector<int> ans;for( ; 1 ; ++i){//找中間點while(A[i][j] < A[i][j+1])j++;while(A[i][j] < A[i][j - 1])j--;//判斷上下是否也滿足if(A[i][j] > A[i + 1][j] && A[i][j] > A[i - 1][j]){ans.push_back(i);ans.push_back(j);return ans;}}return {-1,-1};} };

100% 數據通過測試
總耗時 2399 ms
您的提交打敗了 39.40% 的提交!

總結

以上是生活随笔為你收集整理的LintCode 390. 找峰值 II的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。