LeetCode-034-在排序數組中查找元素的第一個和最後一個位置

在排序數組中查找元素的第一個和最後一個位置

題目描述：給定一個按照升序排列的整數數組 nums，和一個目標值 target。找出給定目標值在數組中的開始位置和結束位置。

若是數組中不存在目標值 target，返回 [-1, -1]。

進階：

• 你能夠設計並實現時間複雜度爲 O(log n) 的算法解決此問題嗎？

示例說明請見LeetCode官網。

來源：力扣（LeetCode）
連接：https://leetcode-cn.com/probl...
著做權歸領釦網絡全部。商業轉載請聯繫官方受權，非商業轉載請註明出處。

解法一：二分查找、遞歸

用left和right記錄開始位置和結束位置，用二分查找法查找target在nums中的位置，從0到nums.length-1分別記爲low和high，mid表示中間值，具體過程以下：

• 計算mid爲low和high的中間值；
• 若是mid的值和target相等，則判斷mid左邊的元素是否跟target相等，若是相等，則遞歸查找開始位置，從low到mid-1查找，若是不相等，則left爲mid；判斷mid右邊的元素是否跟target相等，若是相等，則遞歸查找結束位置，從mid+1到high查找，若是不相等，則right爲mid；
• 若是mid的值大於target，則high更新爲mid-1；
• 若是mid的值小於target，則low更新爲mid+1。

查找完成後返回left和right即爲最終結果。

import javafx.util.Pair;

public class LeetCode_034 {
    public static int[] searchRange(int[] nums, int target) {
        Pair<Integer, Integer> result = find(nums, 0, nums.length - 1, target);
        return new int[]{result.getKey(), result.getValue()};
    }

    public static Pair<Integer, Integer> find(int[] nums, int low, int high, int target) {
        int mid, left = -1, right = -1;
        while (low <= high) {
            mid = (low + high) / 2;
            if (nums[mid] == target) {
                if (mid + 1 <= high && nums[mid + 1] == nums[mid]) {
                    right = find(nums, mid + 1, high, target).getValue();

                } else {
                    right = mid;
                }
                if (mid - 1 >= low && nums[mid - 1] == nums[mid]) {
                    left = find(nums, low, mid - 1, target).getKey();
                } else {
                    left = mid;
                }
                break;
            } else if (nums[mid] > target) {
                high = mid - 1;
            } else {
                low = mid + 1;
            }
        }
        return new Pair<>(left, right);
    }

    public static void main(String[] args) {
        int[] nums = new int[]{5, 7, 7, 8, 8, 10};
        int[] ints = searchRange(nums, 8);
        System.out.println("[" + ints[0] + "," + ints[1] + "]");
    }
}

【每日寄語】 陽春三月，萬物復甦，願全部美好如期而至。