leetcode5938. 找出数组排序后的目标下标(周赛)

找出数组排序后的目标下标


力扣链接

题目描述

给你一个下标从 0 开始的整数数组 nums 以及一个目标元素 target

目标下标 是一个满足 nums[i] == target 的下标 i

nums非递减 顺序排序后,返回由 nums 中目标下标组成的列表。如果不存在目标下标,返回一个 列表。返回的列表必须按 递增 顺序排列。

示例 1:

输入:nums = [1,2,5,2,3], target = 2
输出:[1,2]
解释:排序后,nums 变为 [1,2,2,3,5] 。
满足 nums[i] == 2 的下标是 1 和 2 。

示例 2:

输入:nums = [1,2,5,2,3], target = 3
输出:[3]
解释:排序后,nums 变为 [1,2,2,3,5] 。
满足 nums[i] == 3 的下标是 3 。

示例 3:

输入:nums = [1,2,5,2,3], target = 5
输出:[4]
解释:排序后,nums 变为 [1,2,2,3,5] 。
满足 nums[i] == 5 的下标是 4 。

示例 4:

输入:nums = [1,2,5,2,3], target = 4
输出:[]
解释:nums 中不含值为 4 的元素。

提示:

  • 1 <= nums.length <= 100
  • 1 <= nums[i], target <= 100

解题思路

排序+二分查找

扫描二维码关注公众号,回复: 13495730 查看本文章

代码

	List<Integer> list = new ArrayList<>();

    public List<Integer> targetIndices(int[] nums, int target) {
    
    
        Arrays.sort(nums);
        binarySearch(nums, 0, nums.length - 1, target);
        return list;
    }

    void binarySearch(int[] nums, int low, int high, int target) {
    
    

        while (low <= high) {
    
    
            int avg = low + high >> 1;
            if (nums[avg] == target) {
    
    
                list.add(avg);
                int index = avg - 1;
                while (index >= 0 && nums[index] == target) {
    
    
                    list.add(0, index);
                    index--;
                }
                index = avg + 1;
                while (index < nums.length && nums[index] == target) {
    
    
                    list.add(index);
                    index++;
                }
                break;
            } else if (nums[avg] > target) {
    
    
                high = avg - 1;
            } else {
    
    
                low = avg + 1;
            }
        }
    }

时间复杂度

O(nlog(n))

猜你喜欢

转载自blog.csdn.net/qq_43478625/article/details/121589656