LeetCode 744
Find Smallest Letter Greater Than Target
class Solution {
public:
char nextGreatestLetter(vector<char>& letters, char target) {
sort(letters.begin(), letters.end());
vector<char> arr;
arr.push_back(letters[0]);
char temp = letters[0];
for (int i = 1; i < letters.size(); i++) {
if(letters[i] != temp) {
arr.push_back(letters[i]);
temp = letters[i];
}
}
int low = 0, high = arr.size()-1, mid, flag = 0, res;
while(low<=high) {
mid = low+(high-low)/2;
if (arr[mid] == target) {
flag = 1;
break;
} else if (arr[mid]>target) {
high = mid-1;
} else {
low = mid+1;
}
}
if (flag == 1)
res = (mid+1)%arr.size();
else
res = low%arr.size();
return arr[res];
}
};