cses/cses_1625.cpp at main · EzElephant/cses · GitHub

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
106
107
108
109
110
111
112
113
114
115
116
117
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2,bmi,bmi2,lzcnt,popcnt")
#include <bits/extc++.h>
using namespace std;
const int dir[4][2] = {1, 0, -1, 0, 0, 1, 0, -1};
queue<pair<int, int>> q;
string buffer;
int ans;
bool grid[9][9];

// bool legal(int x, int y)
// {
//     grid[x][y] = true;

//     bool visited[7][7], result = true;
//     for (int i = 0; i < 7; i++)
//         for (int j = 0; j < 7; j++)
//             visited[i][j] = grid[i][j];
//     q.emplace(6, 0);
//     visited[6][0] = true;
//     while (!q.empty())
//     {
//         auto tmp = q.front();
//         q.pop();
//         for (int d = 0; d < 4; d++)
//         {
//             int i = tmp.first + dir[d][0];
//             int j = tmp.second + dir[d][1];
//             if (i >= 0 && j >= 0 && i < 7 && j < 7 && !visited[i][j])
//             {
//                 q.emplace(i, j);
//                 visited[i][j] = true;
//             }
//         }
//     }
//     for (int i = 0; i < 7; i++)
//         for (int j = 0; j < 7; j++)
//             if (visited[i][j] == false)
//                 result = false;

//     grid[x][y] = false;
//     return result;
// }

bool legal(int x, int y)
{
    bool d, u, r, l;
    d = grid[x + 1][y];
    u = grid[x - 1][y];
    r = grid[x][y + 1];
    l = grid[x][y - 1];
    if ((d && u && !r && !l) || (!d && !u && r && l))
        return false;
    return true;
}

void dfs(int step, int x, int y)
{
    if (x == 7 && y == 1 && step < 48)
        return;
    if (step == 48)
    {
        ans++;
        return;
    }

    int s = 0;
    switch (buffer[step])
    {
    case 'D':
        s += 1 << 0;
        break;
    case 'U':
        s += 1 << 1;
        break;
    case 'R':
        s += 1 << 2;
        break;
    case 'L':
        s += 1 << 3;
        break;
    default:
        s += 15;
    }
    for (int d = 0; s; s ^= d)
    {
        d = s & -s;
        int i = x + dir[__lg(d)][0];
        int j = y + dir[__lg(d)][1];
        if (i > 0 && j > 0 && i <= 7 && j <= 7 && !grid[i][j] && legal(i, j))
        {
            grid[i][j] = true;
            dfs(step + 1, i, j);
            grid[i][j] = false;
        }
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);

    cin >> buffer;
    for (int i = 0; i < 9; i++)
    {
        grid[i][8] = true;
        grid[i][0] = true;
        grid[0][i] = true;
        grid[8][i] = true;
    }
    grid[1][1] = true;
    dfs(0, 1, 1);
    cout << ans << '\n';

    return 0;
}