코딩테스트/그래프|트리
[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;
}