给定两个整数 n 和 k,你需要实现一个数组,这个数组包含从 1 到 n 的 n 个不同整数,同时满足以下条件:
① 如果这个数组是 [a1, a2, a3, ... , an] ,那么数组 [|a1 - a2|, |a2 - a3|, |a3 - a4|, ... , |an-1 - an|] 中应该有且仅有 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
提示:
n 和 k 满足条件 1 <= k < n <= 104.
思路:因为我们要保证当且仅当只有k个不同的差值的绝对值,我们知道k+1个数能够得到k个差,因此我们可以先取出前k+1个数进行构造,剩下的数按顺序加入即可。但是我们应该怎样构造呢? 在演草纸上写了写发现每次两头取正好可以构造不同的差值的绝对值,好了,知道怎么构造之后就简单了,因为我们知道后面不用构造的数的差值固定为1,而前k+1个数中一定会存在一个差值为1,但是第k+2个数与已经构造的数列的最后一个数之差肯定是没有出现过的,因此,数量刚好满足题意。
class Solution {
public int[] constructArray(int n, int k) {
int[] ans=new int[n];
int l=1,r=k+1,len=0;
while(l<r) {
ans[len++]=l;
ans[len++]=r;
l++;r--;
}
if((k+1)%2>0) ans[len++]=l;
for(int i=k+2;i<=n;i++)
ans[len++]=i;
return ans;
}
}
袁珂晨 发布了967 篇原创文章 · 获赞 175 · 访问量 21万+ 关注