题目描述
思路
找规律
当k=1时,将1-n按照[1, 2,…, n]的顺序排列,相邻的差均为1,满足k=1的条件。
当k=n-1时,将1-n按照[1, n, 2, n-1,…]的顺序排列,相邻的差此部分n-1开始,依次递减。所以1到n-1的差值均出现一次,满足k=n-1的条件。
由此可以推出,列表的前半部分相邻差均为1,后半部分的相邻差从k开始逐渐递减到1,这样1到k的差值均出现依次,对应的列表为:[1, 2,…, n-k, n, n-k+1, n-1, n-k+2,…]。
Python实现
class Solution:
def constructArray(self, n: int, k: int) -> List[int]:
ans = list(range(1, n-k))
i, j = n-k, n
while i <= j:
ans.append(i)
if i != j:
ans.append(j)
i, j = i+1, j-1
return ans
Java实现
class Solution {
public int[] constructArray(int n, int k) {
int[] ans = new int[n];
int idx = 0;
for (int i = 1; i < n-k; ++i) {
ans[idx] = i;
++idx;
}
for (int i = n-k, j = n; i <= j; ++i, --j) {
ans[idx] = i;
++idx;
if (i != j) {
ans[idx] = j;
++idx;
}
}
return ans;
}
}
C++实现
class Solution {
public:
vector<int> constructArray(int n, int k) {
vector<int> ans;
for (int i = 1; i < n-k; ++i) {
ans.push_back(i);
}
for (int i = n-k, j = n; i <= j; ++i, --j) {
ans.push_back(i);
if (i != j) {
ans.push_back(j);
}
}
return ans;
}
};