#include <bits/stdc++.h>
using namespace std;
int n, m, res, t[3001];
bool yes()
{
vector <bool> dp(n + 1, 0);
for(int i = 2; i <= n; i++)
{
for(int j = 1; j < i; j++)
{
if(t[i] == t[j] && (j == 1 || dp[j - 1] == 1))
{
dp[i] = 1;
break;
}
}
}
return dp[n];
}
void komb(int l)
{
if(l > n)
{
if(yes())
res++;
return;
}
for(int i = 1; i <= m; i++)
{
t[l] = i;
komb(l + 1);
}
}
int main()
{
cin >> n >> m;
komb(1);
cout << res;
}
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 | #include <bits/stdc++.h> using namespace std; int n, m, res, t[3001]; bool yes() { vector <bool> dp(n + 1, 0); for(int i = 2; i <= n; i++) { for(int j = 1; j < i; j++) { if(t[i] == t[j] && (j == 1 || dp[j - 1] == 1)) { dp[i] = 1; break; } } } return dp[n]; } void komb(int l) { if(l > n) { if(yes()) res++; return; } for(int i = 1; i <= m; i++) { t[l] = i; komb(l + 1); } } int main() { cin >> n >> m; komb(1); cout << res; } |
English