#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; } |