力扣链接
题目描述
给你一个下标从 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
解题思路
排序+二分查找
代码
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))