1389. 按既定顺序创建目标数组

题目:

给你两个整数数组 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;
    }
};

耗时和内存使用情况: 

发布了253 篇原创文章 · 获赞 15 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/junjunjiao0911/article/details/105180756