1.题目描述:
给你两个整数n和k,请你构造一个数组,该数组包含从1到n的n个不同正整数,并同时满足下述条件:假设该列表是answer = [a1,a2,a3,... ,an],那么列表[|a1 - a2|,|a2 - a3|,|a3 - a4|,... ,|an-1 - an|]中应该有且仅有k个不同整数。如果存在多种答案,只需返回其中任意一种 。
2.解法:
这道题目需要找到数据排列的特点,即较大的值和较小的值交替排列,能找到题设条件对应的最大k值。详细的思路见图解,图解方法理解之后,代码其实较容易写出。
class Solution {
public int[] constructArray(int n, int k) {
int[] resArr = new int[n];
for(int i = 0;i < n - k - 1;i++){
resArr[i] = i + 1;
}
int oddOreven = (n - k - 1) % 2;
int a = 0;
int b = 0;
for(int i = n - k - 1;i < n;i++){
if(i % 2 == oddOreven){
resArr[i] = n - k + a;
a++;
}else{
resArr[i] = n - b;
b++;
}
}
return resArr;
}
}
3.优化后的代码:最大值与最小值交替排列,等差放在最后
class Solution {
public int[] constructArray(int n, int k) {
int[] ret = new int[n];
ret[0] = 1;
for (int i = 1, interval = k; i <= k; i++, interval--) {
ret[i] = i % 2 == 1 ? ret[i - 1] + interval : ret[i - 1] - interval;
}
for (int i = k + 1; i < n; i++) {
ret[i] = i + 1;
}
return ret;
}
}