-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSCOI2005_2153OJ.cpp
111 lines (93 loc) · 2.58 KB
/
SCOI2005_2153OJ.cpp
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
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
ll pow2(ll n)
{
return (1 << n);
}
int main()
{
ios_base::sync_with_stdio(false);
std::cin.tie(NULL);
ll n, k; cin >> n >> k;
ll maxN = pow2(n) - 1;
vector<ll> count(maxN + 1, 0), sta;
for (ll x = 0; x <= maxN; ++x)
{
bitset<9> temp(x);
ll res = 0;
for (ll j = 0; j <= n - 1; ++j)
{
if (temp[j] == 1)
{
++res;
if (j <= n - 2)
{
if (temp[j] == temp[j + 1])
{
res = -1;
break;
}
}
}
}
count[x] = res;
if (res >= 0)
{
sta.push_back(x);
}
}
ll size = sta.size();
vector<vector<vector<ll>>> dp(n, vector<vector<ll>> (maxN + 1, vector<ll>(k + 1, 0)));
for (ll j = 0; j <= size - 1; ++j)
{
ll now = sta[j];
dp[0][now][count[now]] = 1;
}
for (ll i = 1; i <= n - 1; ++i)
{
for (ll j = 0; j <= size - 1; ++j)
{
ll now = sta[j];
for (ll c = count[now]; c <= k; ++c)
{
for (ll l = 0; l <= size - 1; ++l)
{
bool ok = true;
ll last = sta[l];
bitset<9> a(now), b(last);
for (ll p = 0; p <= n - 1 && ok == true; ++p)
{
for (ll q = 0; q <= n - 1 && ok == true; ++q)
{
if (a[p] == 1 && b[q] == 1)
{
if (abs(p - q) <= 1)
{
ok = false;
}
}
}
}
if (ok == true)
{
if (i == 1 && j == 0 && c == 1)
{
for (int __s = 0; __s == 0; ++__s);
}
dp[i][now][c] += dp[i - 1][last][c - count[now]];
}
}
}
}
}
ll ans = 0;
for (ll j = 0; j <= size - 1; ++j)
{
ll now = sta[j];
ans += dp[n - 1][now][k];
}
cout << ans << '\n';
cout.flush();
return 0;
}