본문 바로가기

Algorithm/Binary Search

(C++) - LeetCode (easy) 744. Find Smallest Letter Greater Than Target

반응형

https://leetcode.com/problems/find-smallest-letter-greater-than-target/description/

 

Find Smallest Letter Greater Than Target - LeetCode

Can you solve this real interview question? Find Smallest Letter Greater Than Target - You are given an array of characters letters that is sorted in non-decreasing order, and a character target. There are at least two different characters in letters. Retu

leetcode.com

이분탐색으로 해결한 문제였습니다.

📕 풀이방법

📔 풀이과정

* n이 10^4이므로 O(n)으로도 풀어도 상관없습니다. c++ 기본 내장 함수 upper_bound를 사용해 구현의 편의성을 도모했습니다.이분탐색으로 target초과하는 최소 iterator를 ans에 저장해줍니다.

 

📔 정답 출력 | 반환

1. ans가 letters.end()라면 해당 값을 가진 원소가 없으므로 첫 글자를 반환합니다.

2. 이외의 경우 iterator의 주소값을 반환해줍니다.


📕 Code

📔 C++

class Solution {
public:
    char nextGreatestLetter(vector<char>& letters, char target) {
        auto ans = upper_bound(letters.begin(), letters.end(), target);
        if(ans == letters.end()) return letters[0];
        return *ans;
    }
};

*더 나은 내용을 위한 지적, 조언은 언제나 환영합니다.