알고리즘/DFS
[백준] 3980-선발 명단(C++)
잉읭응
2021. 1. 24. 17:20
반응형
문제링크:www.acmicpc.net/problem/3980
3980번: 선발 명단
각각의 테스트 케이스에 대해서, 모든 포지션의 선수를 채웠을 때, 능력치의 합의 최댓값을 출력한다. 항상 하나 이상의 올바른 라인업을 만들 수 있다.
www.acmicpc.net
dfs문제입니다.
전형적인 dfs문제입니다. 백트래킹에 속하는 이유는 문제조건중에서
각 선수는 능력치가 0인 포지션에 배치될 수 없다.
라는 조건을 수행해야하기 때문입니다.
일반적인 dfs 방식을 사용해서 구현했습니다.
아래는 정답코드입니다.
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int t = 0, result = 0;
int arr[11][11] = { 0, };
bool visited[11];
void dfs(int cnt, int sum) {
if (cnt == 11) {
result = max(result, sum);
return;
}
for (int i = 0; i < 11; i++) {
if (arr[cnt][i] == 0)
continue;
if (visited[i] == 0) {
visited[i] = 1;
dfs(cnt + 1, sum + arr[cnt][i]);
visited[i] = 0;
}
}
}
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0), cout.tie(0);
cin >> t;
while (t--) {
for (int i = 0; i < 11; i++)
for (int j = 0; j < 11; j++)
cin >> arr[i][j];
memset(visited, 0, sizeof(visited));
result = 0;
dfs(0, 0);
cout << result << "\n";
}
}
반응형