[BOJ/C++] 백준 2798번: 블랙잭

 

🔗 문제 링크 : https://www.acmicpc.net/problem/2798

 

 

첫 제출코드의 채점결과이다.

 

처참한 메모리 사용량(7040KB)을 보고, 코드 개선 욕구가 폭발했다.

 

 

개선 후 채점결과 (만족)

 

---

 

문제

카지노에서 제일 인기 있는 게임 블랙잭의 규칙은 상당히 쉽다. 카드의 합이 21을 넘지 않는 한도 내에서, 카드의 합을 최대한 크게 만드는 게임이다. 블랙잭은 카지노마다 다양한 규정이 있다.

한국 최고의 블랙잭 고수 김정인은 새로운 블랙잭 규칙을 만들어 상근, 창영이와 게임하려고 한다.

김정인 버전의 블랙잭에서 각 카드에는 양의 정수가 쓰여 있다. 그 다음, 딜러는 N장의 카드를 모두 숫자가 보이도록 바닥에 놓는다. 그런 후에 딜러는 숫자 M을 크게 외친다.

이제 플레이어는 제한된 시간 안에 N장의 카드 중에서 3장의 카드를 골라야 한다. 블랙잭 변형 게임이기 때문에, 플레이어가 고른 카드의 합은 M을 넘지 않으면서 M과 최대한 가깝게 만들어야 한다.

N장의 카드에 써져 있는 숫자가 주어졌을 때, M을 넘지 않으면서 M에 최대한 가까운 카드 3장의 합을 구해 출력하시오.

 

 

 

 

 

입출력 예시

입력

첫째 줄에 카드의 개수 N(3 ≤ N ≤ 100)과 M(10 ≤ M ≤ 300,000)이 주어진다. 둘째 줄에는 카드에 쓰여 있는 수가 주어지며, 이 값은 100,000을 넘지 않는 양의 정수이다.

합이 M을 넘지 않는 카드 3장을 찾을 수 있는 경우만 입력으로 주어진다.

 

 

 

출력

첫째 줄에 M을 넘지 않으면서 M에 최대한 가까운 카드 3장의 합을 출력한다.

 

 

 

입출력 예시

 

 

 

---

 

풀이 #1 | 재귀 호출

3개의 수를 조합하여 합을 구하는 재귀 함수를 이용한 풀이다. (재귀 연습할 겸..)

 

재귀함수의 가장 중요한 탈출 조건은 다음과 같다.

• 합이 M을 넘음
• 카드 3장을 모두 선택했을 때

 

재귀함수에 넘겨줄 인자는 다음과 같다.

• start : 현재 선택을 시작한 인덱스
• currSum : 현재까지 선택한 카드의 합
• count : 현재까지 선택된 카드의 개수

#include <iostream>
#include <vector>
using namespace std;

int N, M;
int maxSum = 0;
vector<int> numbers;

void generateCombination(int start, int currSum, int count) {
    /* 합이 M을 넘으면 탈출 */
    if (currSum > M) {
        return;
    }

    /* 3장의 카드를 골랐을 경우 배열에 저장하고 탈출 */
    if (count == 3) {
        if (currSum > maxSum) 
            maxSum = currSum;
        return;
    }

    /* 재귀 호출 */
    for (int i = start; i <= N - 3 + count; ++i) {
        generateCombination(i + 1, currSum + numbers[i], count + 1);
    }
}

int main() {
    /* N, M 입력받기 */
    cin >> N >> M;

    /* N개의 카드들 입력받기 */
    for (int i = 0; i < N; ++i) {
        int num;
        cin >> num;

        numbers.push_back(num);
    }

    generateCombination(0, 0, 0);

    cout << maxSum;
    return 0;
}

위 코드의 채점결과

 

 

 

☁ 개선하기 전 풀이

숫자카드 3개를 조합해서 구한 합들을 굳이 배열에 넣은 풀이다.

조합된 원소들을 눈으로 직접 확인 하려고 작성한 코드인데..  (자연스럽게) 그대로 제출하고 처참한 채점 결과를 마주했다.

 

M을 넘지 않는 합들은 set컨테이너에 정렬되도록 저장하고, M과 가까우면서 가장 큰 수는 set에 저장된 원소들 중에 맨 마지막 값을 추출해 오도록 구현했다.

#include <iostream>
#include <set>
#include <vector>
using namespace std;

int N, M;
vector<int> numbers;
set<int> sums;

void generateCombination(int start, int currSum, int count) {
    /* 합이 M을 넘을경우 탈출 */
    if (currSum > M) {
        return;
    }

    /* 3장의 카드를 골랐을 경우 배열에 저장하고 탈출 */
    if (count == 3) {
        sums.insert(currSum);
        return;
    }

    /* 재귀 호출 */
    for (int i = start; i <= N - 3 + count; ++i) {
        generateCombination(i + 1, currSum + numbers[i], count + 1);
    }
}

int main() {
    /* N, M 입력받기 */
    cin >> N >> M;

    /* N개의 카드들 입력받기 */
    for (int i = 0; i < N; ++i) {
        int num;
        cin >> num;

        numbers.push_back(num);
    }

    generateCombination(0, 0, 0);

    cout << *sums.rbegin();
    return 0;
}

위 코드의 채점결과

 

 

 

풀이 #2 | for문을 이용한 조합

삼중 for문을 이용한 가장 직관적인 풀이라고 생각한다.

#include <stdio.h>
#include <stdlib.h>

int main() {
    int N, M;
    int sum, maxSum = 0;

    /* N, M 입력받기 */
    scanf("%d %d", &N, &M);

    int* cards = (int*)malloc(sizeof(int) * N);
    for (int i = 0; i < N; ++i) {
        scanf("%d", &cards[i]);
    }

    for (int i = 0; i < N - 2; ++i) {
        for (int j = i + 1; j < N - 1; ++j) {
            for (int k = j + 1; k < N; ++k) {
                sum = cards[i] + cards[j] + cards[k];

                if (sum > maxSum && sum <= M)
                    maxSum = sum;
            }
        }
    }

    printf("%d", maxSum);
    free(cards);

    return 0;
}

위 코드의 채점결과

 

---