프로그래머스-로또의 최고 순위와 최저 순위

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

 

 

<풀이>

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int grade(int num){
        if(num == 6) {
           return 1;
            
        }else if(num == 5) {
            return 2;
            
        }else if(num == 4) {
            return 3;
            
        }else if(num == 3) {
            return 4;
            
        }else if(num == 2) {
            return 5;
            
        }else{
            return 6;
            
        }
        
}

vector<int> solution(vector<int> lottos, vector<int> win_nums) {
    vector<int> answer;
    vector<int>::iterator it;
    
    int unknown = 0;
    int eq = 0;

    
    
    for(int i=0; i<lottos.size(); i++){
        it = find(lottos.begin(), lottos.end(),win_nums[i]);
        if(it != lottos.end()){
            eq++;
                
        }  
    }
    
    for(int i=0; i<lottos.size(); i++){
        if(lottos[i] == 0){
                unknown++;
        }
        
    }
        
        
    if(unknown != 0){
        answer.push_back(grade(eq));
        answer.push_back(grade(eq+unknown));
        
        }else{
            answer.push_back(grade(eq));
            answer.push_back(grade(eq));
        }
    
    sort(answer.begin(), answer.end());
    
    return answer;
}