- 重新排列数组
题目描述
给你一个数组 nums ,数组中有 2n 个元素,按 [x1,x2,…,xn,y1,y2,…,yn] 的格式排列。
请你将数组按 [x1,y1,x2,y2,…,xn,yn] 格式重新排列,返回重排后的数组。
示例
示例 1:
输入:nums = [2,5,1,3,4,7], n = 3
输出:[2,3,5,4,1,7]
解释:由于 x1=2, x2=5, x3=1, y1=3, y2=4, y3=7 ,所以答案为 [2,3,5,4,1,7]
示例 2:
输入:nums = [1,2,3,4,4,3,2,1], n = 4
输出:[1,4,2,3,3,2,4,1]
示例 3:
输入:nums = [1,1,2,2], n = 2
输出:[1,2,1,2]
提示:
1 <= n <= 500
nums.length == 2n
1 <= nums[i] <= 10^3
思路
总共有n对(x,y),则每次都直接把x、y分别添加到list里,即第一次循环添加第0和第3个元素,第二次循环添加第1和第4个元素,直到终止。
Python
class Solution(object):
def shuffle(self, nums, n):
list = []
for i in range(1, n+1):
list.append(nums[i-1])
list.append(nums[n+i-1])
return list
"""
:type nums: List[int]
:type n: int
:rtype: List[int]
"""
叮咚
1、range(1, n+1),假设n = 3,则生成一个列表list = [1, 2, 3]。而for x in 列表,是指逐个把列表中的元素取出来,相当于间接实现了i不断+1的效果。
2、错误代码:
class Solution(object):
def shuffle(self, nums, n):
change = []
for i in range(0, len(nums)):
if (i+1) % 2 == 0:
change.insert(i,nums[i+n-1])
else:
change.insert(i,nums[i])
i += 1
return nums
"""
:type nums: List[int]
:type n: int
:rtype: List[int]
"""
for循环内各种变量的变化模拟如下:
根据第一个样例 2 5 1 3 4 7,可知i是从 [0, 6),初始change为[ ],所以首先,i = 0进入循环,根据判断语句可知进入else部分,则添加索引为i的元素,change = [2];随后i = 1再次进入循环,根据判断语句进入if部分,则添加索引为i+n-1的元素,change = [2, 3];随后i = 2再次进入循环,根据判断语句进入else部分,则添加索引为i的元素,change = [2, 3, 1],此时可以发现,change 第三个位置添加的元素应该是5,那么为什么变成1了呢,会发现在i=1的时候,是进入if部分,添加了i+n-1元素,而还没来得及处理索引i=1的元素就进入下一次循环了,即i变成2了,所以代码这样运行下去会不断忽略若干元素。