667. Beautiful Arrangement II

2018/10/05 leetcode 共 913 字,约 3 分钟

title

problem link

Given two integers n and k, you need to construct a list which contains n different positive integers ranging from 1 to n and obeys the following requirement: Suppose this list is [a1, a2, a3, … , an], then the list [|a1 - a2|, |a2 - a3|, |a3 - a4|, … , |an-1 - an|] has exactly k distinct integers.

If there are multiple answers, print any of them.

Example 1:
Input: n = 3, k = 1
Output: [1, 2, 3]

Explanation: The [1, 2, 3] has three different positive integers ranging from 1 to 3, and the [1, 1] has exactly 1 distinct integer: 1.

Example 2:
Input: n = 3, k = 2
Output: [1, 3, 2]

Explanation: The [1, 3, 2] has three different positive integers ranging from 1 to 3, and the [2, 1] has exactly 2

solution

题目大意: 分配1-n,,让隔壁相差的数字有k个不同的

解法: 先分配1-(n-k),然后剩下k+1个数字,从头从尾跳着来,就是一高一低这样

class Solution {
public:
	vector<int> constructArray(int n, int k) {
		vector<int> res;
		for (int i = 1; i < n - k; i++) {
			res.push_back(i);
		}
		for (int i = 0; i <= k; i++) {
			if (i % 2 == 0) {
				res.push_back(n - k + i / 2);
			}
			else {
				res.push_back(n - i / 2);
			}
		}
		return res;
	}
};

Search

    Table of Contents