题目
给你两个整数数组 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
来源:力扣(LeetCode)
思路
先初始化一个数组target,全部用-1填充,然后用下标i循环遍历数组index和nums。
如果target[index[i]]为-1,说明这个位置没有插入过数字,直接复制target[index[i]] = nums[i]即可。
如果target[index[i]]不为-1,说明这个位置已经插入过数字,那么需要要将targer数组的index..length这段元素依次往后移动一位,然后再设置target[index[i]] = nums[i]。
循环次数 nums index target
0 1 0 [1]
1 2 1 [1,2]
2 3 2 [1,2,3]
3 4 3 [1,2,3,4]
4 0 0 [0,1,2,3,4]
实现方式
class Solution(object):
def createTargetArray(self, nums, index):
"""
:type nums: List[int]
:type index: List[int]
:rtype: List[int]
"""
length = len(nums)
#初始化目标target数组
target = [-1] * length
#下标法遍历两个数组
for i in range(length):
#如果target的下标为index[i]的位置没有插入过数字
if target[index[i]] == -1:
#直接复制num[i]给target[index[i]]
target[index[i]] = nums[i]
else:
#将下标为index[i]到length-1处的元素后移
for j in range(length-1, index[i], -1):
target[j] = target[j-1]
#后移完之后在复制target[index[i]]
target[index[i]] = nums[i]
return target
各位大神,有其他思路欢迎留言~
博主:测试生财(一个不为996而996的测开码农)
座右铭:专注测试开发与自动化运维,努力读书思考写作,为内卷的人生奠定财务自由。
内容范畴:技术提升,职场杂谈,事业发展,阅读写作,投资理财,健康人生。
csdn:https://blog.csdn.net/ccgshigao
博客园:https://www.cnblogs.com/qa-freeroad/
51cto:https://blog.51cto.com/14900374
微信公众号:测试生财(定期分享独家内容和资源)