[c++] 프로그래머스 - 로또의 최고 순위와 최저 순위( Level 1)

문제

코딩테스트 연습 - 로또의 최고 순위와 최저 순위

 

코드

#include <string>
#include <vector>
#include <map>
#include <iostream>

using namespace std;

int check(int n){
    if(n<=1) return 6;
    return 7-n;
}
vector<int> solution(vector<int> lottos, vector<int> win_nums) {
    vector<int> answer;
    map<int,int>m;
    int best = 0, worst = 0;
    
    for(int lotto: lottos) m[lotto]++;
    for(int win: win_nums){
        if(m.find(win)!=m.end()) m[win]--;
    }
    
    for(auto it=m.begin(); it!=m.end(); it++){
        if(it->first==0){
            best+=it->second;
            worst+=it->second;
        } 
        else if(it->first!=0){
            if(it->second==0) best++;
            else worst++;
        }
    }
    
    answer.push_back(check(best));
    answer.push_back(check(6-worst));
    return answer;
}

 

풀이

최고로 맞는 개수와 최저로 맞는 개수를 구해가는 식으로 구현했습니다. 

• 0일 때는 worst와 best 전부 ++
• 0이 아닐때는 수가 같을 때 best++, 아니라면 worst--
• 최종적으로 worst는 틀린 개수이기 때문에 맞는 개수로 바꿔주기 위해 6-worst를 통해 순위 계산