프로그래머스[Lv.3] - n + 1 카드게임

문제 설명

당신은 1~n 사이의 수가 적힌 카드가 하나씩 있는 카드 뭉치와 동전 coin개를 이용한 게임을 하려고 합니다. 카드 뭉치에서 카드를 뽑는 순서가 정해져 있으며, 게임은 다음과 같이 진행합니다.

1. 처음에 카드 뭉치에서 카드 n/3장을 뽑아 모두 가집니다. (n은 6의 배수입니다.) 당신은 카드와 교환 가능한 동전 coin개를 가지고 있습니다.
2. 게임은 1라운드부터 시작되며, 각 라운드가 시작할 때 카드를 두 장 뽑습니다. 카드 뭉치에 남은 카드가 없다면 게임을 종료합니다. 뽑은 카드는 카드 한 장당 동전 하나를 소모해 가지거나, 동전을 소모하지 않고 버릴 수 있습니다.
3. 카드에 적힌 수의 합이 n+1이 되도록 카드 두 장을 내고 다음 라운드로 진행할 수 있습니다. 만약 카드 두 장을 낼 수 없다면 게임을 종료합니다.

예를 들어 n = 12, coin = 4이고 [3, 6, 7, 2, 1, 10, 5, 9, 8, 12, 11, 4] 순서대로 카드를 뽑도록 카드 뭉치가 섞여 있습니다.

처음에 3, 6, 7, 2 카드 4장(= n/3)과 동전 4개(= coin)를 가지고 시작합니다. 다음 라운드로 진행하기 위해 내야 할 카드 두 장에 적힌 수의 합은 13(= n+1)입니다. 다음과 같은 방법으로 최대 5라운드까지 도달할 수 있습니다.

1. 1라운드에서 뽑은 카드 1, 10을 동전 두 개를 소모해서 모두 가집니다. 카드 3, 10을 내고 다음 라운드로 진행합니다. 이때 손에 남은 카드는 1, 2, 6, 7이고 동전이 2개 남습니다.
2. 2라운드에서 뽑은 카드 5, 9를 동전을 소모하지 않고 모두 버립니다. 카드 6, 7을 내고 다음 라운드로 진행합니다. 이때 손에 남은 카드는 1, 2고 동전이 2개 남습니다.
3. 3라운드에서 뽑은 카드 8, 12 중 동전 한 개를 소모해서 카드 12를 가집니다. 카드 1, 12을 내고 다음 라운드로 진행합니다. 이때 손에 남은 카드는 2이고 동전이 1개 남습니다.
4. 4라운드에서 뽑은 카드 11, 4 중 동전 한 개를 소모해서 카드 11을 가집니다. 카드 2, 11을 내고 다음 라운드로 진행합니다. 이때 손에 남은 카드와 동전은 없습니다.
5. 5라운드에서 카드 뭉치에 남은 카드가 없으므로 게임을 종료합니다.

처음에 가진 동전수를 나타내는 정수 coin과 카드를 뽑는 순서대로 카드에 적힌 수를 담은 1차원 정수 배열 cards가 매개변수로 주어질 때, 게임에서 도달 가능한 최대 라운드의 수를 return 하도록 solution 함수를 완성해 주세요.

---

 

 

 

 

 

 

제한 사항

• 0 ≤ coin ≤ n
• 6 ≤ cards의 길이 = n < 1,000
  
  • cards[i]는 i+1번째로 뽑는 카드에 적힌 수를 나타냅니다.
  • 1 ≤ cards[i] ≤ n
  • cards의 원소는 중복되지 않습니다.
• n은 6의 배수입니다.

---

 

 

 

 

 

 

풀이

처음에는 DFS로 문제를 풀었는데 시간초과가 났다.

의심은 했는데 역시나였다....

 

정답은 의외로 간단했다.

카드를 받아올 때, 현재 상황이나 미래를 계산할 필요가 없다.

현재 라운드에서 우선 카드 2개를 받아 따로 저장한 뒤, 이번 라운드를 통과할 수 있는지 계산하면 된다.

현재 라운드를 통과하는 경우는 총 3가지이다.

• 현재 들고있는 카드에서 2개로 n+1을 만든다. (사용된 coin: 0)
• 현재 들고있는 카드에서 1개, 현재 라운드에서 받은 카드 중 1개로 n+1을 만든다. (사용된 coin: 1)
• 현재 라운드에서 받은 카드 2개로 n+1을 만든다. (사용된 coin: 2)

즉, 현재 라운드만 통과할 수 있다면 받은 카드를 버릴 필요가 없다.

들고 있다가 필요할 때 사용하면 되는 것이다.

코인이 부족하거나 받아올 카드가 없거나 n+1을 만들 수 있는 경우가 없다면 종료하면 된다. 

---

 

 

 

 

 

전체 코드

#include <string>
#include <vector>
#include <unordered_set>
#include <iostream>

using namespace std;

bool CheckPossible(unordered_set<int>& hands, int target)
{
    bool bPossible = false;
    for(int num : hands)
    {
        if(hands.count(target-num) == 1)
        {
            hands.erase(num);
            hands.erase(target-num);
            bPossible = true;
            
            break;
        }
    }
    
    return bPossible;
}

int solution(int coin, vector<int> cards) {
    int answer = 1;
    
    unordered_set<int> hands;
    unordered_set<int> GetCards;
    
    for(int i = 0; i < cards.size()/3; i++) hands.insert(cards[i]);
    
    int idx = cards.size()/3;
    int target = cards.size()+1;
    
    while(true && idx < cards.size())
    {       
        GetCards.insert(cards[idx]);
        GetCards.insert(cards[idx+1]);
        
        idx+=2;
        
        if(CheckPossible(hands, target))
        {
            answer++;
            continue;
        }
        
        {
            bool bPossible = false;
        
            for(auto num : hands)
            {
                if(GetCards.count(target-num) == 1)
                {
                    hands.erase(num);
                    GetCards.erase(target-num);
                    bPossible = true;
                    
                    break;
                }
            }
            
            if(bPossible && coin >= 1)
            {
                answer++;
                coin--;
                continue;
            }
        }
        
        if(CheckPossible(GetCards, target) && coin >= 2)
        {
            coin -= 2;
            answer++;
            continue;
        }
        
        break;
    }
    
    return answer;
}

---