题目:
给你两个整数数组 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 = [1,2,3,4,0], index = [0,1,2,3,0]
输出:[0,1,2,3,4]
解释:
nums index target
1 0 [1]
2 1 [1,2]
3 2 [1,2,3]
4 3 [1,2,3,4]
0 0 [0,1,2,3,4]
示例 3:
输入:nums = [1], index = [0]
输出:[1]
提示:
1 <= nums.length, index.length <= 100
nums.length == index.length
0 <= nums[i] <= 100
0 <= index[i] <= i
解题思路:
1.一个辅助数组cnt用来记录原数组index中每个元素出现的次数,初始值为-1
2.遍历原数组index,如果当前元素之前没有出现过,即cnt[index[i]]==-1,则将当前nums数组中对应的第i个元素nums[i]放在index[i]处,即res[index[i]]=nums[i];并累加index[i]出现的次数,即cnt[index[i]]++;
3.如果当前元素之前出现过,则将index[i]及其以后的元素都后移一个,空出来index[i]的位置,给新的元素,即
for(int j=s-2;j>=index[i];--j){
res[j+1]=res[j];
}
res[index[i]]=nums[i];
代码实现:
class Solution {
public:
vector<int> createTargetArray(vector<int>& nums, vector<int>& index) {
int s=nums.size();
int cnt[101]={-1};
vector<int> res(s);
for(int i=0;i<s;++i){
if(cnt[index[i]]>=0){//后移元素
for(int j=s-2;j>=index[i];--j){
res[j+1]=res[j];
}
res[index[i]]=nums[i];
}
else{
res[index[i]]=nums[i];
cnt[index[i]]++;
}
}
return res;
}
};
耗时和内存使用情况: