#include <bits/stdc++.h> using namespace std; #define ll long long #define ld long double int A[1000000 + 10]; int R[1000000 + 10]; bool check(int n) { for (int i = 0; i <= n; i++) R[i] = 0; int i = 1, j = n; while (A[i] == 0) i++; while (A[j] == 0) j--; if (i == j) { if (A[i] == 1) return true; else return false; } for (int k = i; k <= j; k++) { if (A[k] == 0) return false; } for (; i < j; i++) { R[i] = A[i] - R[i - 1]; if (R[i] <= 0 && i < j - 1) { return false; } } if (abs(R[j - 1] - A[j]) > 1) return false; else return true; } void solve() { int n; cin >> n; for (int i = 1; i <= n; i++) cin >> A[i]; if (check(n)) cout << "TAK\n"; else { reverse(A + 1, A + n + 1); if (!check(n)) cout << "NIE\n"; else cout << "TAK\n"; } } int main () { ios_base::sync_with_stdio(NULL); cin.tie(0); int t; cin >> t; while (t--) solve(); 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 | #include <bits/stdc++.h> using namespace std; #define ll long long #define ld long double int A[1000000 + 10]; int R[1000000 + 10]; bool check(int n) { for (int i = 0; i <= n; i++) R[i] = 0; int i = 1, j = n; while (A[i] == 0) i++; while (A[j] == 0) j--; if (i == j) { if (A[i] == 1) return true; else return false; } for (int k = i; k <= j; k++) { if (A[k] == 0) return false; } for (; i < j; i++) { R[i] = A[i] - R[i - 1]; if (R[i] <= 0 && i < j - 1) { return false; } } if (abs(R[j - 1] - A[j]) > 1) return false; else return true; } void solve() { int n; cin >> n; for (int i = 1; i <= n; i++) cin >> A[i]; if (check(n)) cout << "TAK\n"; else { reverse(A + 1, A + n + 1); if (!check(n)) cout << "NIE\n"; else cout << "TAK\n"; } } int main () { ios_base::sync_with_stdio(NULL); cin.tie(0); int t; cin >> t; while (t--) solve(); return 0; } |