【数组与矩阵】——leetcode667.优美的排列 II

leetcode667. 优美的排列 II

难度:中等

给定两个整数 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

题目来源：力扣（LeetCode）

思路

这边k是奇数或者偶数的情况这样思考
1.把一个数插入到有序递增1的序列中的时候如果是在中间,那么这个插入的数和它左右两边的数会产生两个不同的差,而原来有序递增1的序列中会有一个为1的差，所以此时产生的k为奇数

2.如果把要插入的数添加到有序递增1的序列的末尾，那么这个插入的数和前面一个数会产生一个差，加上原来有序递增1的序列中会有一个为1的差，所以此时产生的k为偶数

代码实现

	public int[] constructArray(int n, int k) {
		int[] res = new int[n]; // 定义一个数组用来返回结果
		int curNum = 1; // 表示当前要填入数组中的数字
		int needInsert = k / 2; 	// 表示需要插入的元素的数量
		int index = k % 2 == 0 ? n - 1 : n - 2; 	// 根据k是奇数还是偶数判断第一个插入的数的位置
		while (curNum <= needInsert) {	// 1.第一个while循环把需要插入的数填入数组
			res[index] = curNum;
			curNum++;
			index -= 2;	 // 由于插入数每次需要间隔一个位置,所以index-=2
		}
		index = 0;	// 2.从数组的第0个位置开始把递增1的序列填入数组中
		while (curNum < n + 1) {
			if (res[index] == 0) {	// 如果为0表示这个位置上原来没有填过数，就把递增1的序列的数填入
				res[index] = curNum;
				curNum++;
			}
			index++;
		}
		return res;
	}