LeetCode/cpp/find-smallest-letter-greater-than-target.cpp

#include <algorithm>
#include <cassert>
#include <vector>

class Solution {
  public:
    char nextGreatestLetter(const std::vector<char> &letters, char target) {
        auto it = std::lower_bound(letters.begin(), letters.end(), target + 1);
        return it == letters.end() ? letters.front() : *it;
    }
};

int main() {
    Solution s;

    assert((s.nextGreatestLetter(std::vector{'c', 'f', 'j'}, 'a') == 'c'));
    assert((s.nextGreatestLetter(std::vector{'c', 'f', 'j'}, 'c') == 'f'));
    assert((s.nextGreatestLetter(std::vector{'x', 'x', 'y', 'y'}, 'z') == 'x'));

    return 0;
}