와이유스토리

[DFS 순열] 프로그래머스 양궁대회 C++ 본문

코딩테스트/그래프|트리

[DFS 순열] 프로그래머스 양궁대회 C++

유(YOO) 2023. 1. 21. 16:45
#include <string>
#include <vector>

using namespace std;

bool isFind;
int maximum;
int visited[11];
vector<int> answer(11, 0);

void dfs(int depth, int num, vector<int> info) {
    if (num == 0) {
        int lScore = 0;
        int aScore = 0;
        for(int i=0; i<11; i++) {
            if (visited[i] > info[i]) lScore += (10-i);
            else if (visited[i] < info[i]) aScore += (10-i);
            else if (visited[i] == 0) continue; // 라이언, 어피치 둘 다 0개
            else aScore += (10-i); // 같은 화살 개수면 어피치 점수
        }
        if (lScore > aScore) { // 라이언이 이기는 경우
            isFind = true;
            if ((lScore - aScore) > maximum) { // 최대 점수 차이
                for(int i=0; i<11; i++) answer[i] = visited[i];
                maximum = (lScore - aScore);
            }
            else if ((lScore - aScore) == maximum) { // 가장 낮은 점수를 많이 맞힌 경우
                for(int i=10; i>=0; i--) {
                    if (visited[i] > answer[i]) {
                        for(int j=0; j<11; j++) answer[j] = visited[j];
                        break;
                    }
                    else if (answer[i] > visited[i]) break; // 8번, 18번 TC
                }
            }
        }
        return;
    }
    if (depth == 11) return;
    
    // 점수당 라이언의 화살 개수 = (어피치의 화살 개수 + 1) or 0
    for(int i=num; i>=0; i--) {
        visited[depth] = i;
        dfs(depth + 1, num - i, info);
        visited[depth] = 0;
    }
}

vector<int> solution(int n, vector<int> info) {
    // n개 화살을 10개 순열로 나누고 점수 계산(10점부터)
    dfs(0, n, info);
    
    // 어피치와 점수 비교
    if (!isFind) return vector<int> (1, -1);
    return answer;
}
Comments