더보기
  • [ Question ]

배열 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 함수를 작성해주세요.

  • My Code
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> array, vector<vector<int>> commands) {
    vector<int> answer;
    for(int i = 0 ; i < commands.size(); i++)
    {
        vector<int> temp;
        for(int j = commands[i][0]-1; j < commands[i][1]; j++)
            temp.push_back(array[j]);
        sort(temp.begin(),temp.end());
        answer.push_back(temp[commands[i][2]-1]);
        temp.clear();
    }
    return answer;
}
  • Better Code
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> array, vector<vector<int>> commands) {
    vector<int> answer;
    vector<int> temp;

    for(int i = 0; i < commands.size(); i++) {
        temp = array;
        sort(temp.begin() + commands[i][0] - 1, temp.begin() + commands[i][1]); // 원본 그대로 받은거에서 정렬
        answer.push_back(temp[commands[i][0] + commands[i][2]-2]);	// temp의 정답요소를 찾을 때 -2하는 것이 포인트
    }
    return answer;
}

 

  • [ CheckPoint ]
    • 나는 매 과정마다 요소를 찾을 때 필요한 범위만큼의 임시벡터를 만들고, 그것을 정렬하고, 필요한 요소를 찾았다. 
    • 이것도 맞는 생각이지만, 애초에 임시벡터에 원본벡터를 그냥 대입하고, 정렬시 범위를 시작점과 끝나는점을 잘 설정한다.
      이후, 필요한 요소를 찾을 때 인덱싱부분에 -2를 하는 이유는, 1번째 요소는 벡터[0] 과 같음을 생각했을 때
      원래는 필요한 범위만큼의 새로생긴 벡터에서도 1번째 요소는 벡터[0] 과 같음이 적용되므로 그 과정을 생각했을 때 -2인덱싱을 해줘야 알맞은 요소가 정답 벡터에 들어갔었다.
728x90
반응형
더보기

[ Question ]

배열 arr가 주어집니다. 배열 arr의 각 원소는 숫자 0부터 9까지로 이루어져 있습니다. 이때, 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거하려고 합니다. 단, 제거된 후 남은 수들을 반환할 때는 배열 arr의 원소들의 순서를 유지해야 합니다.

  • arr = [1, 1, 3, 3, 0, 1, 1] 이면 [1, 3, 0, 1] 을 return 합니다.
  • arr = [4, 4, 4, 3, 3] 이면 [4, 3] 을 return 합니다.

배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성해 주세요.

제한사항

  • 배열 arr의 크기 : 1,000,000 이하의 자연수
  • 배열 arr의 원소의 크기 : 0보다 크거나 같고 9보다 작거나 같은 정수
  • My Code
#include <vector>
#include <iostream>

using namespace std;

vector<int> solution(vector<int> arr) 
{
    vector<int> answer;
    int index = 0;
    for(int i = 0 ; i < arr.size();i++)
    {
        if(i == arr.size()-1 && arr[i] == arr[i-1])
            answer.insert(answer.begin()+index++,arr[i]);
        else if(arr[i] != arr[i+1])
            answer.insert(answer.begin()+index++,arr[i]);
    }

    return answer;
}
  • Better Code
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;

vector<int> solution(vector<int> arr) 
{

    arr.erase(unique(arr.begin(), arr.end()),arr.end());

    vector<int> answer = arr;
    return answer;
}
  • [ CheckPoint ]
    • 나는 벡터길이만큼 반복시키면서 현재 인덱스요소와 다음 인덱스요소 다르면 현재 인덱스요소를 정답 벡터에 삽입하는 방식을 생각. 다만 마지막 인덱스요소 일때, 다음 인덱스 검사하면 indexing error발생하기때문에, 마지막 인덱스 일 경우에만, 바로 전 요소와 비교 후 같으면 정답벡터에 삽입했다.
    • vector.erase(unique(v.begin(),v.end()),v.end()) 를 기억하자.
    • <algorithm> 헤더파일 필요
    • unique() = 주변에 요소를 기준으로 중복되는 요소를 제거하기때문에, 반드시 정렬 후 사용해라.
    • 실행 후, 중복요소들은 벡터 맨뒤로 간다. 실행 완료 후, 버렸던 원소의 첫 시작위치를 반환하므로, erase()함수를 통해 그 주소부터 벡터끝까지 필요없는 요소들 삭제
728x90
반응형

+ Recent posts