[BaekJoon 2798번] 블랙잭(C++)

※주의※

저의 풀이가 정답은 아닙니다.

다른 코드가 더 효율적이거나 좋을 수 있습니다.

언제나 다른 사람의 코드는 참고만 하시기 바랍니다.

 

 

[문제 풀이 사이트]

 

 

2798번: 블랙잭

 

 

[문제 설명]

 

카지노에서 제일 인기 있는 게임 블랙잭의 규칙은 상당히 쉽다. 카드의 합이 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장의 합을 출력한다.

 

 

[입출력 예]

 

 

입력	출력
5 21 5 6 7 8 9	21
10 500 93 181 245 214 315 36 185 138 216 295	497

 

 

[문제 풀이]

 

주어진 카드 개수 중에서 3개를 뽑아 고른 카드의 합이 M을 넘지 않으면서 M과 가까운 값이 되어야 한다.

쉬운 문제라 생각이 되어 처음에 next_permutation을 이용하여 조합 알고리즘을 쓰지 않고, 재귀로 풀려고 했을 때 시간초과가 발생했다.

 

(시간 초과 코드)

#include <bits/stdc++.h>
using namespace std;

int N, M; // 카드 개수, 고른 카드의 합이 M을 넘지 않아야하고 M과 최대한 가깝게 만들어야 함
vector<int> card;
vector<bool> selected;
int cardlen;
int ans;

void allsearch(int idx, int cnt) {

    //3 장의 카드를 골랐을 때
    if(cnt == 3) {
        int sum = 0;
        for(int i = 0; i < cardlen; ++i) {
            if(selected[i]) {
                sum += card[i];
                if (sum > M) break;
            }
        }

        if(sum <= M) ans = max(ans, sum);
    }

    for(int i = idx; i < cardlen; ++i) {
        if(selected[i]) continue;
        selected[i] = true;
        allsearch(i, cnt + 1);
        selected[i] = false;
    }
}

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(nullptr); cout.tie(nullptr);
    cin >> N >> M;
    for(int i = 0; i < N; ++i) {
        int n;
        cin >> n;
        card.push_back(n);
        selected.push_back(false);
    }
    cardlen = card.size();
    ans = 0;
    allsearch(0, 0); // 아직 아무것도 선택하지 않음
    
    cout << ans;
    return 0;
}

 

 

그래서 시간을 줄일 수 있는 방법을 몇분 고민하다가 안되서, next_permutation을 이용하여 조합 알고리즘으로 구현하여 제출했다.

다풀고나서 다른사람들의 풀이를 보는데 주로 3중 for문으로 풀어서 제출한 사람이 많았다.

그리고 여러 블로그 글을 읽어보니 나와 같이  재귀로 풀어서 시간초과가 발생해 틀린 사람들이 있었다.

 

 

[최종 코드]

 

 

[GitHub]

 

 

GitHub - SmallPeanutPark/BAEKJOON: Study BaekJoon for Coding Test

 

 

#include <bits/stdc++.h>
using namespace std;

int N, M; // 카드 개수, 고른 카드의 합이 M을 넘지 않아야하고 M과 최대한 가깝게 만들어야 함

int main(void) {
    ios::sync_with_stdio(false);
    cin.tie(nullptr); cout.tie(nullptr);
    vector<int> card;
    int ans = 0;
    cin >> N >> M;

    for(int i = 0; i < N; ++i) {
        int n;
        cin >> n;
        card.push_back(n);
    }

    sort(card.begin(), card.end());

    vector<bool> selected(N);
    fill(selected.end() - 3, selected.end(), true);

    do {
        int sum = 0;
        for(int i = 0; i < N; ++i) {
            if(selected[i]) {
                sum += card[i];
                if(sum > M) break;
            }
        }
        if(sum <= M) ans = max(ans, sum);
    } while(next_permutation(selected.begin(), selected.end()));

    cout << ans;

    return 0;
}