#include <iostream> #include <cmath> using namespace std; bool IsCorect(int* trees, int n) { if (trees[0] == trees[n - 1] && n != 1) return true; for (int i = n - 3; i > 0; i--) { if (trees[i] == trees[0]) { return IsCorect(&trees[i + 1], n - i - 1); } } return false; } void Add(int* numbers, int m) { numbers[0]++; if (numbers[0] > m) { numbers[0] = 1; Add(&numbers[1], m); } } int main() { int n, m; cin >> n >> m; int trees[n]; int answer = 0; for (int i = 0; i < n; i++) { trees[i] = 1; } for (int i = 1; i <= pow(m, n); i++) { if(IsCorect(trees, n)) { answer++; } Add(trees, m); } cout << answer % (1000000000 + 7); return 0; }
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 | #include <iostream> #include <cmath> using namespace std; bool IsCorect(int* trees, int n) { if (trees[0] == trees[n - 1] && n != 1) return true; for (int i = n - 3; i > 0; i--) { if (trees[i] == trees[0]) { return IsCorect(&trees[i + 1], n - i - 1); } } return false; } void Add(int* numbers, int m) { numbers[0]++; if (numbers[0] > m) { numbers[0] = 1; Add(&numbers[1], m); } } int main() { int n, m; cin >> n >> m; int trees[n]; int answer = 0; for (int i = 0; i < n; i++) { trees[i] = 1; } for (int i = 1; i <= pow(m, n); i++) { if(IsCorect(trees, n)) { answer++; } Add(trees, m); } cout << answer % (1000000000 + 7); return 0; } |