对于某些固定的 N,如果数组 A 是整数 1, 2, …, N 组成的排列,使得:对于每个 i < j,都不存在 k 满足 i < k < j 使得 A[k] * 2 = A[i] + A[j]。那么数组 A 是漂亮数组。
给定 N,返回任意漂亮数组 A(保证存在一个)。
解题思路:
关于漂亮数组的性质:
首先,不存在 k 满足 i < k < j 使得 A[k] * 2 = A[i] + A[j]。
时间复杂度:O(NlogN)
空间复杂度:O(NlogN)
class Solution:
def beautifulArray(self, N: int) -> List[int]:
ans=[1]
while len(ans)<N: # 退出时长度可能超过N,可能存在比N大的数
ans = [2*i for i in ans] + [2*j-1 for j in ans]
return [x for x in ans if x <=N ]