[LeetCode] 162. Find Peak Element 풀이

Problem

Key Idea

이진탐색으로 \(nums[m]\) 값과 양 옆의 값을 비교해준다.

1. \(nums[m-1] \gt nums[m]\) 이면:
   index \(m\) 왼쪽에 \(peak\)가 존재.

   r <- m;
   

2. \(nums[m] \lt nums[m+1]\) 이면:
   index \(m\) 오른쪽에 \(peak\)가 존재.

   l <- m+1;
   

3. 둘다 아니면: index \(m\) 자체가 \(peak\).

   return m;
   

Implementation

/**
 * written: 2021. 12. 29. Wed. 16:05:01 [UTC+9]
 * jooncco의 mac에서.
 **/

typedef vector<int> vi;

class Solution {
public:
    int findPeakElement(vi &nums) {
        int n= nums.size();
        // nums[-1]과 nums[n]이 -∞ 이므로 size가 1이면 자체가 peak
        if (n == 1) return 0;
        
        int l= 0, r= n-1;
        while (l < r) {
            int m= l+(r-l)/2;
            if (m > 0 && nums[m] < nums[m-1]) r= m;
            else if (m < n-1 && nums[m] < nums[m+1]) l= m+1;
            else return m;  // peak를 찾았으니 바로 return.
        }
        // l == peak index
        return l;
    }
};

• Time: \(O(\log n)\)