[프로그래머스 / C++] 전국 대회 선발 고사

문제

전국 대회 선발 고사 : 문제 링크

문제 분석

• 0번부터 n - 1번까지 n명의 학생 중 3명을 선발하는 전국 대회 선발 고사를 보았다. 등수가 높은 3명을 선발해야 하지만, 개인 사정으로 전국 대회에 참여하지 못하는 학생들이 있어 참여가 가능한 학생 중 등수가 높은 3명을 선발하기로 했다. 각 학생들의 선발 고사 등수를 담은 정수 배열 rank와 전국 대회 참여 가능 여부가 담긴 boolean 배열 attendance가 매개변수로 주어진다. 전국 대회에 선발된 학생 번호들을 등수가 높은 순서대로 각각 a, b, c번이라고 할 때 10000 x a + 100 x b + c를 return
• pair 클래스를 사용하고, attended의 원소를 오름차순 정렬하기 위해 sort() 함수를 사용하므로 algorithm 헤더를 include
• 전국 대회에 선발된 학생 번호로 연산한 값을 저장할 answer을 0으로, 참여 가능한 학생들의 등수와 번호를 저장할 pair형 벡터 attended를 초기화. for loop를 통해 rank의 첫번째부터 마지막까지 순환하고, if문을 사용하여 참여여부가 true인 경우 학생의 등수와 번호를 attended에 저장. loop 탈출 후 sort() 함수를 통해 등수 오름차순대로 정렬. 1등 학생의 번호 10000 + 2등 학생의 번호 100 + 3등 학생의 번호를 연산한 값을 answer에 저장하고 return

pair 클래스 사용법
template <class T1, class T2> struct pair;

• T1 : first, T2 : second
• p.first : p의 첫번째 인자를 return
• p.second : p의 두번째 인자를 return

algorithm 헤더의 sort() 함수 사용법
void sort(T start, T end, Compare comp); //comp 인자가 공란이면 오름차순 정렬

• sort(v.begin(), v.end(), compare); // 사용자 정의 함수 사용
• sort(v.begin(), v.end(), greater<자료형>()); // 내림차순
• sort(v.begin(), v.end(), less<자료형>()); // 오름차순

풀이

#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<int> rank, vector<bool> attendance) {
    int answer = 0;
    vector<pair<int, int>> attended;
    
    for(int i = 0; i < rank.size(); ++i) {
        if(attendance[i]) attended.push_back({rank[i], i});
    }
    sort(attended.begin(), attended.end());
    answer = 10000 * attended[0].second + 100 * attended[1].second + attended[2].second;
    return answer;
}