题目:
给你一个下标从 1 开始、包含 不同 整数的数组 nums
,数组长度为 n
。
你需要通过 n
次操作,将 nums
中的所有元素分配到两个数组 arr1
和 arr2
中。在第一次操作中,将 nums[1]
追加到 arr1
。在第二次操作中,将 nums[2]
追加到 arr2
。之后,在第 i
次操作中:
- 如果
arr1
的最后一个元素 大于arr2
的最后一个元素,就将nums[i]
追加到arr1
。否则,将nums[i]
追加到arr2
。
通过连接数组 arr1
和 arr2
形成数组 result
。例如,如果 arr1 == [1,2,3]
且 arr2 == [4,5,6]
,那么 result = [1,2,3,4,5,6]
。
返回数组 result
。
思路:
题目说要把nums中所有元素分配到两个数组arr1和arr2中,那么nums中第一个数和第二个数一定是先确定分配好的,所以先将nums中第一个数和第二个数分配到arr1和arr2,在比较arr1和arr2最后一个数的大小,再来确定nums[i]存放在arr1还是arr2。
代码:
class Solution:
def resultArray(self, nums: List[int]) -> List[int]:
arr1 = [nums[0]]
arr2 = [nums[1]]
for i in range(2,len(nums)):
if arr1[-1] > arr2 [-1]:
arr1.append(nums[i])
else:
arr2.append(nums[i])
return arr1 + arr2
心得:题目比较基础,忽略题目“第一句下标从1开始”,直接做就好了
原题链接:https://leetcode.cn/problems/distribute-elements-into-two-arrays-i/description/