#include <iostream> #include <vector> using namespace std; long long gcd(long long int a, long long int b) { if (b == 0) return a; return gcd(b, a % b); } long long lcm(long long a, long long b) { if (a > b) return (a / gcd(a, b)) * b; else return (b / gcd(a, b)) * a; } int calculate_offset(int n, vector<vector<vector<int>>>& masters, int node) { int offset_sum = 0; for (int i = 0; i < masters[node].size(); i++) { int l_m = masters[node][i][0]; // l_m int p_s = masters[node][i][1]; // p_s int offset = (n - p_s + l_m - 1) / l_m; offset_sum += offset; } return offset_sum; } long long combined_reset_time(int master_node, int l_m, int l_s, int p_s, vector<vector<int>>& outputs, vector<vector<vector<int>>>& masters) { int k = l_m / l_s + 1; int surplus = k * l_s - l_m; int min_reset_time = -1; int s_o = 0; //int first = surplus * l_m - /*s_o(n)*/ + p_s - l_m + 1; //int first_mod = first % l_m; //int reset_time = first + (l_m - (first % l_m) % l_m); // add binary search int a, b; for (int i = 1; i <= 100; i++) { int o = calculate_offset(i, masters, master_node); int p_s_o = (p_s - o + l_m) % l_m; int o_s = (i - p_s_o + l_m - 1) / l_m; int L = o_s - o; int R = surplus; if ((i - o + l_m) % l_m == 0 && (i - o_s + l_s) % l_s == 0) { min_reset_time = i; break; } } return min_reset_time * 2; } void print_vector(vector<int>& v) { for (auto i : v) { cout << i << " "; } cout << endl; } int main() { int n; cin >> n; vector<vector<int>> outputs(n + 1); vector<int> counters(n + 1); int outputs_number, output_to; for (int i = 1; i <= n; i++) { cin >> outputs_number; for (int j = 0; j < outputs_number; j++) { cin >> output_to; outputs[i].push_back(output_to); } } if (outputs[1].size() == 0) { cout << 1; return 0; } vector<vector<vector<int>>> masters(n + 1); unsigned long long counter = 0; unsigned long long suits = 0; int currentNode, nextNode; int increment; int newCounter; do { currentNode = 1; while (outputs[currentNode].size() > 0) { nextNode = outputs[currentNode][counters[currentNode]]; newCounter = (counters[currentNode] + 1) % outputs[currentNode].size(); counter += newCounter - counters[currentNode]; counters[currentNode] = newCounter; currentNode = nextNode; } suits++; //cout << counter << endl; } while (counter != 0); cout << suits << endl; 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 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 118 119 120 121 122 123 124 | #include <iostream> #include <vector> using namespace std; long long gcd(long long int a, long long int b) { if (b == 0) return a; return gcd(b, a % b); } long long lcm(long long a, long long b) { if (a > b) return (a / gcd(a, b)) * b; else return (b / gcd(a, b)) * a; } int calculate_offset(int n, vector<vector<vector<int>>>& masters, int node) { int offset_sum = 0; for (int i = 0; i < masters[node].size(); i++) { int l_m = masters[node][i][0]; // l_m int p_s = masters[node][i][1]; // p_s int offset = (n - p_s + l_m - 1) / l_m; offset_sum += offset; } return offset_sum; } long long combined_reset_time(int master_node, int l_m, int l_s, int p_s, vector<vector<int>>& outputs, vector<vector<vector<int>>>& masters) { int k = l_m / l_s + 1; int surplus = k * l_s - l_m; int min_reset_time = -1; int s_o = 0; //int first = surplus * l_m - /*s_o(n)*/ + p_s - l_m + 1; //int first_mod = first % l_m; //int reset_time = first + (l_m - (first % l_m) % l_m); // add binary search int a, b; for (int i = 1; i <= 100; i++) { int o = calculate_offset(i, masters, master_node); int p_s_o = (p_s - o + l_m) % l_m; int o_s = (i - p_s_o + l_m - 1) / l_m; int L = o_s - o; int R = surplus; if ((i - o + l_m) % l_m == 0 && (i - o_s + l_s) % l_s == 0) { min_reset_time = i; break; } } return min_reset_time * 2; } void print_vector(vector<int>& v) { for (auto i : v) { cout << i << " "; } cout << endl; } int main() { int n; cin >> n; vector<vector<int>> outputs(n + 1); vector<int> counters(n + 1); int outputs_number, output_to; for (int i = 1; i <= n; i++) { cin >> outputs_number; for (int j = 0; j < outputs_number; j++) { cin >> output_to; outputs[i].push_back(output_to); } } if (outputs[1].size() == 0) { cout << 1; return 0; } vector<vector<vector<int>>> masters(n + 1); unsigned long long counter = 0; unsigned long long suits = 0; int currentNode, nextNode; int increment; int newCounter; do { currentNode = 1; while (outputs[currentNode].size() > 0) { nextNode = outputs[currentNode][counters[currentNode]]; newCounter = (counters[currentNode] + 1) % outputs[currentNode].size(); counter += newCounter - counters[currentNode]; counters[currentNode] = newCounter; currentNode = nextNode; } suits++; //cout << counter << endl; } while (counter != 0); cout << suits << endl; return 0; } |