[프로그래머스/C++] Level1_로또의 최고 순위와 최저 순위

<2021 Dev-Matching : 웹 백엔드 개발자>

 

-문제는 더보기에 있어요.-

<문제 설명>

로또 6/45(이하 '로또'로 표기)는 1부터 45까지의 숫자 중 6개를 찍어서 맞히는 대표적인 복권입니다. 아래는 로또의 순위를 정하는 방식입니다. 

순위	당첨 내용
1	6개 번호가 모두 일치
2	5개 번호가 일치
3	4개 번호가 일치
4	3개 번호가 일치
5	2개 번호가 일치
6(낙첨)	그 외

로또를 구매한 민우는 당첨 번호 발표일을 학수고대하고 있었습니다. 하지만, 민우의 동생이 로또에 낙서를 하여, 일부 번호를 알아볼 수 없게 되었습니다. 당첨 번호 발표 후, 민우는 자신이 구매했던 로또로 당첨이 가능했던 최고 순위와 최저 순위를 알아보고 싶어 졌습니다.
알아볼 수 없는 번호를 0으로 표기하기로 하고, 민우가 구매한 로또 번호 6개가 44, 1, 0, 0, 31 25라고 가정해보겠습니다. 당첨 번호 6개가 31, 10, 45, 1, 6, 19라면, 당첨 가능한 최고 순위와 최저 순위의 한 예는 아래와 같습니다.

당첨 번호	31	10	45	1	6	19	결과
최고 순위 번호	31	0→10	44	1	0→6	25	4개 번호 일치, 3등
최저 순위 번호	31	0→11	44	1	0→7	25	2개 번호 일치, 5등

• 순서와 상관없이, 구매한 로또에 당첨 번호와 일치하는 번호가 있으면 맞힌 걸로 인정됩니다.
• 알아볼 수 없는 두 개의 번호를 각각 10, 6이라고 가정하면 3등에 당첨될 수 있습니다.
  • 3등을 만드는 다른 방법들도 존재합니다. 하지만, 2등 이상으로 만드는 것은 불가능합니다.
• 알아볼 수 없는 두 개의 번호를 각각 11, 7이라고 가정하면 5등에 당첨될 수 있습니다.
  • 5등을 만드는 다른 방법들도 존재합니다. 하지만, 6등(낙첨)으로 만드는 것은 불가능합니다.

민우가 구매한 로또 번호를 담은 배열 lottos, 당첨 번호를 담은 배열 win_nums가 매개변수로 주어집니다. 이때, 당첨 가능한 최고 순위와 최저 순위를 차례대로 배열에 담아서 return 하도록 solution 함수를 완성해주세요.

제한사항

• lottos는 길이 6인 정수 배열입니다.
• lottos의 모든 원소는 0 이상 45 이하인 정수입니다.
  • 0은 알아볼 수 없는 숫자를 의미합니다.
  • 0을 제외한 다른 숫자들은 lottos에 2개 이상 담겨있지 않습니다.
  • lottos의 원소들은 정렬되어 있지 않을 수도 있습니다.
• win_nums은 길이 6인 정수 배열입니다.
• win_nums의 모든 원소는 1 이상 45 이하인 정수입니다.
  • win_nums에는 같은 숫자가 2개 이상 담겨있지 않습니다.
  • win_nums의 원소들은 정렬되어 있지 않을 수도 있습니다.

---

입출력 예

lottos	win_nums	result
[44, 1, 0, 0, 31, 25]	[31, 10, 45, 1, 6, 19]	[3, 5]
[0, 0, 0, 0, 0, 0]	[38, 19, 20, 40, 15, 25]	[1, 6]
[45, 4, 35, 20, 3, 9]	[20, 9, 3, 45, 4, 35]	[1, 1]

입출력 예 설명

입출력 예 #1
문제 예시와 같습니다.

입출력 예 #2
알아볼 수 없는 번호들이 아래와 같았다면, 1등과 6등에 당첨될 수 있습니다.

당첨 번호	39	19	20	40	15	25	결과
최고 순위 번호	0→38	0→19	0→20	0→40	0→15	0→25	6개 번호 일치, 1등
최저 순위 번호	0→21	0→22	0→23	0→24	0→26	0→27	0개 번호 일치, 6등

입출력 예 #3
민우가 구매한 로또의 번호와 당첨 번호가 모두 일치하므로, 최고 순위와 최저 순위는 모두 1등입니다.

---

1. 실제로 사용되는 로또 순위의 결정 방식과는 약간 다르지만, 이 문제에서는 지문에 명시된 대로 로또 순위를 결정하도록 합니다. 

 

 

<문제 풀이>

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

using namespace std;

vector<int> solution(vector<int> lottos, vector<int> win_nums) {
    vector<int> answer;
    vector<int> result(12);
    int count_zero = 0;
    int count_cor = 0;

    sort(lottos.begin(), lottos.end());
    sort(win_nums.begin(), win_nums.end());

    while(lottos[count_zero] == 0) count_zero++;

    count_cor = set_intersection(lottos.begin(), lottos.end(), 
                                 win_nums.begin(), win_nums.end(), 
                                 result.begin()) - result.begin();

    if((count_zero + count_cor) < 2) answer.push_back(6);
    else answer.push_back(7 - (count_zero + count_cor));

    if(count_cor == 0) count_cor++;
    answer.push_back(7 - count_cor);

    return answer;
}

아마 이 문제를 인터넷에 검색하면 많은 사람들이 2중 for문을 이용하여 풀었을거라고 생각이 든다. 근데 난 얼마전에 집합연산 함수를 보고 꼭 써먹어보고 싶어서... 교집합을 구하는 연산을 이용하여 문제를 해결했다.

 

내가 짜려고 한 로직은,

1. lottos 배열에서 0인 원소의 갯수를 구한다. -> count_zero

2. lottos와 win_nums 중 교집합인 것의 원소 개수를 구한다. -> count_cor

3. 최고로 맞추는 로또 번호 개수는 count_zero + count_cor

4. 최저로 맞추는 로또 번호 개수는 count_cor

위 수들을 이용하여 답을 도출해냈다.

 

1번은 while문을 이용하여 lottos를 선형탐색 하였다.

2번에서 집합연산을 사용하였는데 <algorithm> 헤더파일에 내장되어있는 set_intersection 함수이다.

이 함수를 쓰기 전 연산에 넣을 vector는 정렬이 되어있는 상태여야 한다.

연산에 넣을 vector의 범위에 맞는 iterator 4가지를 적어주고 마지막 인자로는 미리 공간을 확보해둔 벡터의 첫 iterator 값을 넣어준다. 이 함수의 리턴 값은 교집합에 해당하는 값 저장이 끝난 지점의 iterator 값이다. 

즉, 이 반환된 result의 iterator 값에서 result.begin()을 빼주면 교집합에 해당하는 원소의 개수를 알 수 있는 것이다.

(-> 여기서 포인터와 iterator가 다르다는 것을 느꼈다. 포인터였으면 int형 크기에 따라 sizeof(int) 로 나누어 주었을텐데..)

 

이렇게 구한 뒤 최고, 최저 몇등인지를 체크할 때 주의할 점이 있다. (이 때문인지 case 15에서 계속 실패가 떴다ㅠ)

나는 6개 맞추면 1등, 5개 맞추면 2등 이렇기에 7에서 맞춘 원소의 개수를 빼면 등수가 나올 수 있도록 코드를 짰는데 이 경우에는 다 틀린 경우에 7등이라는 결과를 내뱉게 된다. 그 경우에 대한 조건만 추가하니 바로 코드가 통과되었다.

---

<문제 후기>

내가 쓴 코드에서는 시간 복잡도가 sort 함수에서 O(N * log(N)) , count_zero 를 셀 때 O(N), set_intersection 함수를 사용할 때 O(N) 가 된다. 

다른 사람들처럼 2중 for문을 사용하게 되면 O(N^2) 이 되는데 사실 이 문제는 커봐야 N이 6이라서 속도 면에서 큰 차이가 없었다. 만약에 후에 값이 커지는 걸 생각한다면 내가 쓴 풀이가 좀 더 좋지 않을까 하지만............. 사실 그렇게 큰 차이를 낼 거 같진 않....

 

그래도 최근에 공부한 걸 직접 써보니 더 복습이 된 기분이었다. 남들과는 다르게 푼 풀이가 뭔가 짜릿하달까 ㅎㅎ; 이상한 곳에 짜릿함을 느끼는 편..

 

번외) 요즘 정보처리기사 공부중인데 하루종일 글만 보다가 코테 좀 푸니까 오히려 머리가 맑아지는 기분이다. 사실 코딩 진짜 싫어하는데 이런 내 모습 보면 좀.. 소름;