leetcode-在排序数组中查找元素的第一个和最后一个位置

110 阅读 0 评论 73 点赞

我是靠谱客的博主鲤鱼云朵，这篇文章主要介绍leetcode-在排序数组中查找元素的第一个和最后一个位置，现在分享给大家，希望可以做个参考。

题目：给定一个按照升序排列的整数数组 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]

题解：实现一个查找目标数字在数组中第一次出现的位置，之后两次调用该函数，一次调用查找target在数组中第一次出现的位置，第二次调用查找target+1在数组中第一次出现的位置，则target在数组中最后一次出现的位置为target+1第一次出现位置的前一个位置，注意在查找时需要考虑数组长度为1的情况，所以h需要设置为nums.length。

复制代码

package com.ljq.sort;

import java.awt.Label;
import java.util.Scanner;

/** 
* @author 作者 Your-Name: 
* @version 创建时间：2019年8月29日 上午11:14:59 
* 类说明 
*/
public class FindFirstAndLast {

public static void main(String[] args) {
		// TODO 自动生成的方法存根
		Scanner scanner = new Scanner(System.in);
		
		while(scanner.hasNext()) {
			String[] strings = scanner.nextLine().split(" ");
			int len = strings.length;
			int[] nums = new int[len-1];
			int target = Integer.parseInt(strings[0]);
			
			for(int i=0;i<len-1;i++) {
				nums[i] = Integer.parseInt(strings[i+1]);
			}
			
			
			int[] ret = searchRange(nums, target);
			for(int x:ret) {
				System.out.print(x+" ");
			}
			System.out.println();
		}
		
		scanner.close();
	}
	
	public static int[]  searchRange(int[] nums,int target) {
		int[] ret = new int[] {-1,-1};
		int first = searchRangeCore(nums, target);
		int end = searchRangeCore(nums, target+1)-1;
		
		if(first<nums.length&&end<nums.length&&nums[first]==target&&nums[end]==target) {
			ret[0] = first;
			ret[1] = end;
		}
		return ret;
		
	}
	
	public static int searchRangeCore(int[] nums,int target) {
		/*
		 * 这里的边界值是考虑数组长度为1的情况
		 */
		int l=0,h=nums.length;
		while(l<h) {
			int m = l+(h-l)/2;
			if(nums[m]>=target) {
				h = m;
			}else {
				l = m+1;
			}
		}
		return l;
	}

}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package com.ljq.sort;

import java.awt.Label;
import java.util.Scanner;

/** 
* @author 作者 Your-Name: 
* @version 创建时间：2019年8月29日 上午11:14:59 
* 类说明 
*/
public class FindFirstAndLast {

	public static void main(String[] args) {
		// TODO 自动生成的方法存根
		Scanner scanner = new Scanner(System.in);
		
		while(scanner.hasNext()) {
			String[] strings = scanner.nextLine().split(" ");
			int len = strings.length;
			int[] nums = new int[len-1];
			int target = Integer.parseInt(strings[0]);
			
			for(int i=0;i<len-1;i++) {
				nums[i] = Integer.parseInt(strings[i+1]);
			}
			
			
			int[] ret = searchRange(nums, target);
			for(int x:ret) {
				System.out.print(x+" ");
			}
			System.out.println();
		}
		
		scanner.close();
	}
	
	public static int[]  searchRange(int[] nums,int target) {
		int[] ret = new int[] {-1,-1};
		int first = searchRangeCore(nums, target);
		int end = searchRangeCore(nums, target+1)-1;
		
		if(first<nums.length&&end<nums.length&&nums[first]==target&&nums[end]==target) {
			ret[0] = first;
			ret[1] = end;
		}
		return ret;
		
	}
	
	public static int searchRangeCore(int[] nums,int target) {
		/*
		 * 这里的边界值是考虑数组长度为1的情况
		 */
		int l=0,h=nums.length;
		while(l<h) {
			int m = l+(h-l)/2;
			if(nums[m]>=target) {
				h = m;
			}else {
				l = m+1;
			}
		}
		return l;
	}

}