#include<bits/stdc++.h> using namespace std; const int lim = 1'000'003, maint = 1'000'000'003; int tab[lim]; int dp[lim]; bool dpb[lim]; int powr[lim]; bool powrb[lim]; int dp2[lim]; bool dp2b[lim]; int powr2[lim]; bool powr2b[lim]; int main(){ std::ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int t; cin >> t; for(int licz = 0; licz < t; licz++){ int n; cin >> n; for(int i = 1; i <= n; i++){ cin >> tab[i]; } bool czy_lew = 0, czy_cos = 0, coska = 0, ost = 0; dp[0] = -1; int l = -1, r = -1; for(int i = 1; i <= n; i++){ if(l == -1 && tab[i] >= 1) l = i; if(tab[i] >= 1) r = i; if(tab[i] >= 1 && coska) ost = 1; if(l != -1 && tab[i] == 0) coska = 1; if(tab[i] <= dp[i - 1]) dp[i] = -1; else dp[i] = tab[i] - 1 - dp[i - 1] - 1; powr[i] = tab[i] - 1 - min(dp[i - 1], tab[i] - 1); } dp2[n + 1] = -1; for(int i = n; i >= 1; i--){ if(tab[i] <= dp2[i + 1]) dp2[i] = -1; else dp2[i] = tab[i] - 1 - dp2[i + 1] - 1; powr2[i] = tab[i] - 1 - min(dp2[i + 1], tab[i] - 1); } bool win = 0; dpb[l - 1] = 1; powrb[l - 1] = 1; for(int i = l; i <= r; i++){ // cout << dp[i] << " " << dp2[i] << ' ' << powr[i] << ' ' << powr2[i] << "\n"; dpb[i] = dpb[i - 1] & (dp[i] >= 0); powrb[i] = powrb[i - 1] & (powr[i] >= 0); } dp2b[r + 1] = 1; powr2b[r + 1] = 1; for(int i = r; i >= l; i--){ dp2b[i] = dp2b[i + 1] & (dp2[i] >= 0); powr2b[i] = powr2b[i + 1] & (powr2[i] >= 0); } for(int i = l; i <= r; i++){ // cout << dp[i] << " " << dp2[i] << ' ' << powr[i] << ' ' << powr2[i] << "\n"; if(dpb[i] && dp2b[i]){ // if(((- dp[i] - dp2[i] - 1) == -tab[i])) cout << i << "p1 \n"; win |= ((-dp[i] - dp2[i] - 1) == -tab[i]); } if(dpb[i] && powr2b[i] && dp2b[i]){ // if(((-dp[i] - powr2[i] - 1) == -tab[i])) cout << i << "p \n"; win |= ((-dp[i] - powr2[i] - 1) == -tab[i]); } if(powrb[i] && dp2b[i] && dpb[i]){ // if(((-dp2[i] - powr[i] - 1) == -tab[i])) cout << i << "p2 \n"; win |= ((-dp2[i] - powr[i] - 1) == -tab[i]); } dp[i] = 0; dpb[i] = 0; dp2[i] = 0; dp2b[i] = 0; powr[i] = 0; powrb[i] = 0; powr2[i] = 0; powr2b[i] = 0; } if(win && !ost) cout << "TAK\n"; else cout << "NIE\n"; } }
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 | #include<bits/stdc++.h> using namespace std; const int lim = 1'000'003, maint = 1'000'000'003; int tab[lim]; int dp[lim]; bool dpb[lim]; int powr[lim]; bool powrb[lim]; int dp2[lim]; bool dp2b[lim]; int powr2[lim]; bool powr2b[lim]; int main(){ std::ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int t; cin >> t; for(int licz = 0; licz < t; licz++){ int n; cin >> n; for(int i = 1; i <= n; i++){ cin >> tab[i]; } bool czy_lew = 0, czy_cos = 0, coska = 0, ost = 0; dp[0] = -1; int l = -1, r = -1; for(int i = 1; i <= n; i++){ if(l == -1 && tab[i] >= 1) l = i; if(tab[i] >= 1) r = i; if(tab[i] >= 1 && coska) ost = 1; if(l != -1 && tab[i] == 0) coska = 1; if(tab[i] <= dp[i - 1]) dp[i] = -1; else dp[i] = tab[i] - 1 - dp[i - 1] - 1; powr[i] = tab[i] - 1 - min(dp[i - 1], tab[i] - 1); } dp2[n + 1] = -1; for(int i = n; i >= 1; i--){ if(tab[i] <= dp2[i + 1]) dp2[i] = -1; else dp2[i] = tab[i] - 1 - dp2[i + 1] - 1; powr2[i] = tab[i] - 1 - min(dp2[i + 1], tab[i] - 1); } bool win = 0; dpb[l - 1] = 1; powrb[l - 1] = 1; for(int i = l; i <= r; i++){ // cout << dp[i] << " " << dp2[i] << ' ' << powr[i] << ' ' << powr2[i] << "\n"; dpb[i] = dpb[i - 1] & (dp[i] >= 0); powrb[i] = powrb[i - 1] & (powr[i] >= 0); } dp2b[r + 1] = 1; powr2b[r + 1] = 1; for(int i = r; i >= l; i--){ dp2b[i] = dp2b[i + 1] & (dp2[i] >= 0); powr2b[i] = powr2b[i + 1] & (powr2[i] >= 0); } for(int i = l; i <= r; i++){ // cout << dp[i] << " " << dp2[i] << ' ' << powr[i] << ' ' << powr2[i] << "\n"; if(dpb[i] && dp2b[i]){ // if(((- dp[i] - dp2[i] - 1) == -tab[i])) cout << i << "p1 \n"; win |= ((-dp[i] - dp2[i] - 1) == -tab[i]); } if(dpb[i] && powr2b[i] && dp2b[i]){ // if(((-dp[i] - powr2[i] - 1) == -tab[i])) cout << i << "p \n"; win |= ((-dp[i] - powr2[i] - 1) == -tab[i]); } if(powrb[i] && dp2b[i] && dpb[i]){ // if(((-dp2[i] - powr[i] - 1) == -tab[i])) cout << i << "p2 \n"; win |= ((-dp2[i] - powr[i] - 1) == -tab[i]); } dp[i] = 0; dpb[i] = 0; dp2[i] = 0; dp2b[i] = 0; powr[i] = 0; powrb[i] = 0; powr2[i] = 0; powr2b[i] = 0; } if(win && !ost) cout << "TAK\n"; else cout << "NIE\n"; } } |