LeetCode 667. Beautiful Arrangement II

Discription

给定两个整数n和k，构建一个n个元素数组，数组要满足一下要求：
假设数组[a1,a2....an]那么[|a2-a1|,|a3-a2|.....]包含k个不同的整数。如果有多个数组输出任意一个数组。1<=k<n<=10000。

Example

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

Solution

给定n那么k最大为n-1，假设这k个数字是 n-1,n-2,n-3...1，所以数列可以是1,n-1,2,n-2,....，比如给定n=9 k=8则数列可以是1 9 2 8 3 7 6 4 5，可以看出这组数据差值是8 7 6 5 4 3 2 1，如果给定k<n-1，则将剩余的项用递增或者递减顺序即可。

Code

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