(프로그래머스)K번째수

Level1

언어 : C++

---

배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하려 합니다.

예를 들어 array가 [1, 5, 2, 6, 3, 7, 4], i = 2, j = 5, k = 3이라면

1. array의 2번째부터 5번째까지 자르면 [5, 2, 6, 3]입니다.
2. 1에서 나온 배열을 정렬하면 [2, 3, 5, 6]입니다.
3. 2에서 나온 배열의 3번째 숫자는 5입니다.

배열 array, [i, j, k]를 원소로 가진 2차원 배열 commands가 매개변수로 주어질 때, commands의 모든 원소에 대해 앞서 설명한 연산을 적용했을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성해주세요.

---

#include <string>
#include <vector>
#include <algorithm>
#include <iostream>

using namespace std;

vector<int> solution(vector<int> array, vector<vector<int>> commands) {
    vector<int> answer;
    vector<int> cal; // 중간 정렬을 위한 벡터
    int array_size = 0;
    int commands_size = 0;
    int i, k, start, end = 0;
    
    array_size = array.size();
    commands_size = commands.size();
    
    for(i=0; i<commands_size; i++)
    {
        start = commands[i][0] - 1; // start 는 문제에서의 'i'를 뜻함
        end = commands[i][1] - 1; // end 는 문제에서의 'j'를 뜻함
        
        for(k=start; k<=end; k++)
        {
            cal.push_back(array[k]); // start 부터 end 까지의 숫자를 cal 벡터에 삽입
        }
            
        sort(cal.begin(), cal.end()); // cal 벡터 오름차순 정렬
        
        answer.push_back(cal[commands[i][2]-1]); // 'commands[i][2]-1'는 문제에서의 'K번째'를 뜻함
        cal.clear(); // cal 벡터 초기화
    }
    
    return answer;
}

*******************************************************************************************************

딱히 어려운점은 없었던 문제다.

*******************************************************************************************************

 

문제링크 : https://programmers.co.kr/learn/courses/30/lessons/42748?language=cpp