// PA 2024 2B - Wyliczanka #include <iostream> #include <string> #include <vector> #include <algorithm> using namespace std; bool checkSequence(vector<int> *nums, int n, int pos, bool isStart, bool onlyZeroes) { if ((*nums)[pos] < 0) { return false; } if ((*nums)[pos] > 0 && onlyZeroes) { return false; } if (pos == n - 1) { if ((*nums)[pos] <= 1) { return true; } else { return false; } } (*nums)[pos + 1] -= (*nums)[pos]; if (!isStart) { if (checkSequence(nums, n, pos + 1, false, (*nums)[pos] == 0)) { return true; } } ++(*nums)[pos + 1]; if (checkSequence(nums, n, pos + 1, true, (*nums)[pos] == 0)) { return true; } --(*nums)[pos + 1]; (*nums)[pos + 1] += (*nums)[pos]; return false; } int main() { ios_base::sync_with_stdio(0); int t, n, a; cin >> t; for (int i = 0; i < t; ++i) { vector<int> nums; cin >> n; for (int j = 0; j < n; ++j) { cin >> a; if (a != 0 || nums.size() > 0) { nums.push_back(a); } } while (nums.back() == 0) { nums.pop_back(); } n = nums.size(); if (checkSequence(&nums, n, 0, false, false)) { cout << "TAK" << endl; } else { cout << "NIE" << 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 | // PA 2024 2B - Wyliczanka #include <iostream> #include <string> #include <vector> #include <algorithm> using namespace std; bool checkSequence(vector<int> *nums, int n, int pos, bool isStart, bool onlyZeroes) { if ((*nums)[pos] < 0) { return false; } if ((*nums)[pos] > 0 && onlyZeroes) { return false; } if (pos == n - 1) { if ((*nums)[pos] <= 1) { return true; } else { return false; } } (*nums)[pos + 1] -= (*nums)[pos]; if (!isStart) { if (checkSequence(nums, n, pos + 1, false, (*nums)[pos] == 0)) { return true; } } ++(*nums)[pos + 1]; if (checkSequence(nums, n, pos + 1, true, (*nums)[pos] == 0)) { return true; } --(*nums)[pos + 1]; (*nums)[pos + 1] += (*nums)[pos]; return false; } int main() { ios_base::sync_with_stdio(0); int t, n, a; cin >> t; for (int i = 0; i < t; ++i) { vector<int> nums; cin >> n; for (int j = 0; j < n; ++j) { cin >> a; if (a != 0 || nums.size() > 0) { nums.push_back(a); } } while (nums.back() == 0) { nums.pop_back(); } n = nums.size(); if (checkSequence(&nums, n, 0, false, false)) { cout << "TAK" << endl; } else { cout << "NIE" << endl; } } return 0; } |