题目是LeetCode第181场周赛的第一题,链接:按既定顺序创建目标数组。具体描述为:给你两个整数数组 nums 和 index。你需要按照以下规则创建目标数组:
- 目标数组 target 最初为空。
- 按从左到右的顺序依次读取 nums[i] 和 index[i],在 target 数组中的下标 index[i] 处插入值 nums[i] 。
- 重复上一步,直到在 nums 和 index 中都没有要读取的元素。
请你返回目标数组。题目保证数字插入位置总是存在。
示例1:
输入:nums = [0,1,2,3,4], index = [0,1,2,2,1]
输出:[0,4,1,3,2]
解释:
nums index target
0 0 [0]
1 1 [0,1]
2 2 [0,1,2]
3 2 [0,1,3,2]
4 1 [0,4,1,3,2]
示例2:
输入:nums = [0,1,2,3,4], index = [0,1,2,2,1]
输出:[0,4,1,3,2]
解释:
nums index target
0 0 [0]
1 1 [0,1]
2 2 [0,1,2]
3 2 [0,1,3,2]
4 1 [0,4,1,3,2]
示例3:
输入:nums = [0,1,2,3,4], index = [0,1,2,2,1]
输出:[0,4,1,3,2]
解释:
nums index target
0 0 [0]
1 1 [0,1]
2 2 [0,1,2]
3 2 [0,1,3,2]
4 1 [0,4,1,3,2]
这道题是easy难度的,所以可以直接利用list的add(index, value)操作完成,这么做的时间复杂度是 ,因为平均每次插入需要 ,而一共需要插入n次。
JAVA版代码如下:
class Solution {
public int[] createTargetArray(int[] nums, int[] index) {
int L = index.length;
List<Integer> lst = new LinkedList<Integer>();
for (int i = 0; i < L; ++i) {
lst.add(index[i], nums[i]);
}
int[] result = new int[L];
for (int i = 0; i < L; ++i) {
result[i] = lst.get(i);
}
return result;
}
}
提交结果如下:
结果当然不咋滴,因为还是会频繁涉及到链表的节点插入什么的,所以这里又可以有另一个想法,不使用链表了,直接从index数组出发,得到每个数实际在最终数组中的索引。首先可以给定一个result数组存放最终结果,如果我们从后往前看index数组,会发现对于一个index[i],它最终的真实位置就是在result数组中现在还没有填充数的所有位置中的第index[i]个。有了这个发现,实现起来也就很容易了,当然时间复杂度还是 ,但可以省去频繁插入数值的操作,而只需要遍历数组。
JAVA版代码如下:
class Solution {
public int[] createTargetArray(int[] nums, int[] index) {
int L = index.length;
int[] result = new int[L];
boolean[] fill = new boolean[L];
for (int i = L - 1; i >= 0; --i) {
int count = -1;
for (int j = 0; j < L; ++j) {
if (!fill[j]) {
if (++count == index[i]) {
fill[j] = true;
result[j] = nums[i];
break;
}
}
}
}
return result;
}
}
提交结果如下: