667. 优美的排列 II
中文版的翻译要仔细看,不小心会引导歧路上
给定两个整数 n 和 k,你需要实现一个数组,这个数组包含从 1 到 n 的 n 个不同整数,同时满足以下条件:
① 如果这个数组是 [a1, a2, a3, … , an] ,那么数组 [|a1 - a2|, |a2 - a3|, |a3 - a4|, … , |an-1 - an|] 中应该有且仅有 k 个不同整数;.
此处意思为列表中的元素,有 k 中不同差值,不是等差数列
② 如果存在多种答案,你只需实现并返回其中任意一种.
- 示例 1:
输入: n = 3, k = 1
输出: [1, 2, 3]
解释: [1, 2, 3] 包含 3 个范围在 1-3 的不同整数, 并且 [1, 1] 中有且仅有 1 个不同整数 : 1
- 示例 2:
输入: n = 3, k = 2
输出: [1, 3, 2]
解释: [1, 3, 2] 包含 3 个范围在 1-3 的不同整数, 并且 [2, 1] 中有且仅有 2 个不同整数: 1 和 2
3-1=2 差值为2, 3-2=1,差值为1, 就是2中差值
-
提示:
n 和 k 满足条件 1 <= k < n <= 104.
链接:https://leetcode-cn.com/problems/beautiful-arrangement-ii
题解:
- 初始化为 1 到 n 的 list,此时所有的差值为1,也只有 1 种差值
- 归纳枚举方法,通过找规律列决出某一种符合条件的列表,如 l[k:] 的元素逆序后插入到 l[:k]的每个元素后;
- 翻转元素方法:从第二位到末位开始不停翻转,翻转一次增加一种差值
"""列表翻转——切片: 既每次讲rst[i:]的元素逆序一次
In [8]: r=[1,2,3,4,5,6,7,8]
In [19]: r[2:]
Out[19]: [3, 4, 5, 6, 7, 8]
In [18]: r[:1:-1]
Out[18]: [8, 7, 6, 5, 4, 3]
In [20]: r[2:] = r[:1:-1]
In [21]: r
Out[21]: [1, 2, 8, 7, 6, 5, 4, 3]
"""
class Solution:
def constructArray(self, n: int, k: int) -> List[int]:
rst = list(range(1,n+1))
for i in range(1,k):
rst[i:] = rst[:i-1:-1] # 翻转,既每次讲rst[i:]的元素逆序一次,
return rst