LeetCode 667. 优美的排列 II（找规律）

示例 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 <= 10^4.

class Solution {
public:
    vector<int> constructArray(int n, int k) {
        int i;
        vector<int> ans(n);
        for(i = 0; i < n; ++i)
        	ans[i] = i+1;
        for(i = 1; i < k; ++i)
        	reverse(ans,i,n-1);
        return ans;
    }

    void reverse(vector<int>& arr, int i, int j)
    {
    	while(i < j)
    		swap(arr[i++], arr[j--]);
    }
};

class Solution {
public:
    vector<int> constructArray(int n, int k) {
        int i, d = n-1;
        bool flag = true;
        vector<int> ans(n,1);
        for(i = 1; i < k; ++i,--d)
        {
        	if(flag)
        		ans[i] = ans[i-1]+d;
        	else
        		ans[i] = ans[i-1]-d;
        	flag = !flag;
        }
        d = 1;
        for( ; i < n; ++i)
        {
        	if(flag)
        		ans[i] = ans[i-1]+d;
        	else
        		ans[i] = ans[i-1]-d;
        }
        return ans;
    }
};