BOJ

[백준] 15684번 사다리조작 - C++

구현, 백트래킹, 브루트포스

Posted by dongjune on November 20, 2020

문제

15684 사다리조작

풀이

사다리 게임이 주어지면, 가로선을 추가해서 모든 번호가 자신의 번호로 도착하게 만들어야 한다.
이 때 추가하는 가로선의 최소 개수를 구하는 문제이다.
이 문제 역시 dfs를 통해 선을 추가하는 모든 경우의 수를 검사했다.

소스 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#include <iostream>
#include <vector>
#define min(a, b) a < b ? a : b

using namespace std;

int n, m, h;
vector<vector<int>> line(11, vector<int>(31));
int ans = 10;

// 입력
void input()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    cin >> n >> m >> h;

    for (int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= h; j++)
        {
            line[i][j] = i;
        }
    }
    int a, b;
    while (m--)
    {
        cin >> a >> b;
        line[b][a] = b + 1;
        line[b + 1][a] = b;
    }
}

// 모든 선이 자신의 번호에 도착하면 true, 아니면 false 반환
bool checkLine()
{
    for (int i = 1; i <= n; i++)
    {
        int cur_line = i;
        for (int j = 1; j <= h; j++)
        {
            if (line[cur_line][j] != cur_line)
                cur_line = line[cur_line][j];
        }
        if (cur_line != i)
            return false;
    }
    return true;
}

// 모든 경우의 수 탐색
void dfs(int L, int index)
{
    if (L > 3 || L >= ans)
    {
        return;
    }
    if (checkLine())
    {
        ans = min(ans, L);
        return;
    }
    for (int i = index; i < n; i++)
    {
        for (int j = 1; j <= h; j++)
        {
            // 이미 다리가 놓아져있는 경우
            if (line[i][j] != i)
                continue;

            int right = i + 1;

            // 오른쪽 세로줄에 놓인 다리가 없다면 다리 놓기
            if (line[right][j] == right)
            {
                line[i][j] = right;
                line[right][j] = i;
                dfs(L + 1, i);
                line[i][j] = i;
                line[right][j] = right;
            }
        }
    }
}

void solve()
{
    if (checkLine())
        cout << 0 << '\n';
    else
    {
        dfs(0, 1);
        ans = ans == 10 ? -1 : ans;
        cout << ans << '\n';
    }
}

int main()
{
    input();
    solve();

    return 0;
}