1. 程式人生 > 其它 >#點分治,樹狀陣列#洛谷 5311 [Ynoi2011] 成都七中

#點分治,樹狀陣列#洛谷 5311 [Ynoi2011] 成都七中

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

難度:簡單

 

給你一個排序後的字元列表 letters ,列表中只包含小寫英文字母。另給出一個目標字母 target,請你尋找在這一有序列表裡比目標字母大的最小字母。

在比較時,字母是依序迴圈出現的。舉個例子:

如果目標字母 target = 'z' 並且字元列表為 letters = ['a', 'b'],則答案返回 'a'
 

示例 1:

輸入: letters = ["c", "f", "j"],target = "a"
輸出: "c"
示例 2:

輸入: letters = ["c","f","j"], target = "c"
輸出: "f"
示例 3:

輸入: letters = ["c","f","j"], target = "d"
輸出: "f"
 

提示:

2 <= letters.length <= 104
letters[i] 是一個小寫字母
letters 按非遞減順序排序
letters 最少包含兩個不同的字母
target 是一個小寫字母

 

思路:

  • 這題標註是簡單,所以大家不要想複雜了
  • 注意:字母是迴圈的出現的,根據例題,也就是說當所有的字元都小於 target, 迴圈一圈之後,回到起點 ,結果就是起點

方法一:線性查詢

/**
 * 線性查詢
 * @param letters
 * @param target
 * @return
 */
public char nextGreatestLetter(char[] letters, char target) {
    for (char letter : letters) {
        if (target < letter) {
            return letter;
        }
    }
    return letters[0];
}

方法二:二分查詢

我們之後,二分法的寫法一般由兩種

  • left < right => left = mid + 1, right = mid
  • left <= right => left = mid + 1, right = mid - 1

如果 letters = ["c", "f", "j"], target = "j" 如果使用第一種,那麼left 最終是等於2,結果就是 letters[left % letters.length] = "f" 如果是第二種,那麼left最終等於3,結果就是 "c" 所以不能使用第一種

/**
 * 二分查詢
 * @param letters
 * @param target
 * @return
 */
public char nextGreatestLetter2(char[] letters, char target) {
   int left = 0, right = letters.length - 1;
   while (left <= right) {
       int mid = left + (right - left) / 2;
       if (letters[mid] > target) {
           right = mid - 1;
       } else {
           left = mid + 1;
       }
   }
   return letters[left % letters.length];
}