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

码农王同学

发布于 2021-01-15 10:55:09

2.2K0

发布于 2021-01-15 10:55:09

文章被收录于专栏：后端Coder后端Coder

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

1，问题描述

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

2，示例描述

示例 1：

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

输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]
示例 3：

输入：nums = [], target = 0
输出：[-1,-1]
 

提示：

0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109

3，题解思路

本题基于我们最熟悉的集合LinkedHashMap键值对集合来做

4，题解程序


import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class SearchRanageTest2 {
    public static void main(String[] args) {
        int[] nums = {3, 3, 3};
        int target = 3;
        
        int[] searchRange = searchRange(nums, target);
        for (int num : searchRange) {
            System.out.print(num + "\t");
        }
    }

    public  static  int[] searchRange(int[] nums, int target) {
        if (nums == null || nums.length == 0) {
            return new int[]{-1, -1};
        }

        Map<Integer, Integer> map = new LinkedHashMap<>(nums.length);
        for (int i = 0; i < nums.length; i++) {
            map.put(i, nums[i]);
        }
        
        List<Map.Entry<Integer, Integer>> collect = map
            .entrySet()
            .stream()
            .filter(x -> x.getValue() == target)
            .collect(Collectors.toList());
        
        if (collect.size() == 0) {
            return new int[]{-1, -1};
        }
        
        if (collect.size() == 1) {
            return new int[]{collect.get(0).getKey(), collect.get(0).getKey()};
        }
   
        if (collect.size() > 2) {
            return new int[]{collect.get(0).getKey(), collect.get(collect.size() - 1).getKey()};
        }
        
        int[] result = new int[collect.size()];
        int index = 0;
        for (Map.Entry<Integer, Integer> entry : collect) {
            result[index++] = entry.getKey();
        }
      
        return result;
    }
}