leetcode 667. Beautiful Arrangement II

来源:互联网 发布:高考志愿填报软件 编辑:程序博客网 时间:2024/05/21 17:30

#include<iostream>

#include<vector>

#define min(a,b) (a>b?b:a)

using namespacestd;

class Solution {

public:

    vector<int> constructArray(int n,int k) {

        int x= k%2==0? k/2+1 : (k+1)/2;

        vector<int> res;

        if(k%2==0)

        {

            res.push_back(x);

            int current_add_number=1;

            int count=1;

            while(count<=k)

            {

                res.push_back(x-current_add_number);

                res.push_back(x+current_add_number);

                current_add_number++;

                count+=2;

            }

            res.pop_back();

        }

        else

        {

            res.push_back(x);

            int current_add_number=1;

            int count=1;

            while(count<k)

            {

                res.push_back(x+current_add_number);

                res.push_back(x-current_add_number);

                current_add_number++;

                count+=2;

            }

        }

        for(int i=k+1;i<=n;i++)

            res.push_back(i);

        return res;

    }

};

一种构造方法


原创粉丝点击