정렬되지 않은 배열 nums가 있다고 가정하고 nums[0]
이 문제를 해결하기 위해 다음 단계를 따릅니다. −
nums와 동일한 요소를 가진 하나의 배열 x를 만듭니다.
x 배열 정렬
i :=x의 크기 – 1, j :=(x – 1의 크기) / 2 및 n :=nums 배열의 크기
0 ~ n – 1 범위의 l에 대해 각 단계에서 l을 2씩 증가
nums[l] :=x[j]
j를 1 감소
범위 1에서 n – 1의 l에 대해 각 단계에서 l을 2씩 증가시킵니다.
nums[l] :=x[i]
i를 1만큼 감소
이해를 돕기 위해 다음 구현을 살펴보겠습니다. −
예시
#include <bits/stdc++.h>
using namespace std;
void print_vector(vector<auto> v){
cout << "[";
for(int i = 0; i<v.size(); i++){
cout << v[i] << ", ";
}
cout << "]"<<endl;
}
class Solution {
public:
void wiggleSort(vector<int>& nums) {
vector <int> x(nums);
sort(x.begin(), x.end());
int i = x.size() - 1 ;
int j = (x.size() - 1)/2;
int n = nums.size();
for(int l = 0; l < n; l +=2){
nums[l] = x[j--];
}
for(int l = 1; l < n; l +=2){
nums[l] = x[i--];
}
}
};
main(){
vector<int> v = {1,5,1,1,6,4};
Solution ob;
(ob.wiggleSort(v));
print_vector(v);
}
입력
[1,5,1,1,6,4]
출력
[1, 6, 1, 5, 1, 4, ]