Java實現 LeetCode 34.在排序數組中查找元素的第一個和最後一個位置(雙指針)

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

你的算法時間複雜度必須是 O(log n) 級別。

如果數組中不存在目標值,返回 [-1, -1]。

示例 1:

輸入: nums = [5,7,7,8,8,10], target = 8
輸出: [3,4]
示例 2:

輸入: nums = [5,7,7,8,8,10], target = 6
輸出: [-1,-1]

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array
著作權歸領釦網絡所有。商業轉載請聯繫官方授權,非商業轉載請註明出處。

解:

  • 這題是使用二分查找,查找左右邊界
    1. 使用二分查找,查找最左邊target的位置
    1. 使用二分查找,查找最右邊target的位置
class Solution {
    public int[] searchRange(int[] nums, int target) {
        int len = nums.length;
        if(len < 1) return new int[]{-1,-1};
		int l = 0;
		int r = len;
		int i = -1, j = -1;
		//查找最左邊的值
		while(l < r) {
			int mid = l + (r - l)/2;
			if(nums[mid] == target) {//查找最左值
				r = mid;
			}else if(nums[mid] < target) {
				l = mid + 1;
			}else if(nums[mid] > target) {
				r = mid;
			}
		}
		if(l < len && nums[l] == target) 
		i = l;
		
		//查找最右邊的值
		l = 0;
		r = len;
		while(l < r) {
			int mid = l + (r - l)/2;
			if(nums[mid] == target) {
				l = mid + 1;//繼續向右找
			}else if(nums[mid] < target) {
				l = mid + 1;
			}else if(nums[mid] > target) {
				r = mid;
			}
		}
		if(l > 0 && nums[l-1] == target)
		j = l - 1;//因爲區間是左閉右開l+1 = mid -> mid = l-1
		
		return new int[]{i,j};
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章