#include<bits/stdc++.h> using namespace std; void debug(vector<int> vec){ for(auto x:vec){ cout << x << " "; } cout << "\n"; } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int z; cin >> z; while(z--){ int n; vector<long long> zabawki; cin >> n; int x; for(int i=0;i<n;i++){ cin >> x; zabawki.push_back(x); } long long sum=0; bool used=0; bool ended=0; for(int x=0;x<n-1;x++){ // if(zabawki[x]==0 && zabawki[x+1]==0) // continue; if(zabawki[x]<0 || (zabawki[x]>0 && ended)){ sum = 1; break; } if(zabawki[x]>=zabawki[x+1] && !used && zabawki[x]!=0){ used=1; zabawki[x]--; } zabawki[x+1]-=zabawki[x];zabawki[x]-=zabawki[x]; if(used && !ended){ if(zabawki[x+1]==0){ ended=1; }else{ zabawki[x+1]--; } } } if(zabawki[n-1]<0)sum=1; if(zabawki[n-1]>1)sum=1; if(zabawki[n-1]==1 && used)sum=1; if(sum==0){ cout << "TAK\n"; }else{ cout << "NIE\n"; } } 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 | #include<bits/stdc++.h> using namespace std; void debug(vector<int> vec){ for(auto x:vec){ cout << x << " "; } cout << "\n"; } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int z; cin >> z; while(z--){ int n; vector<long long> zabawki; cin >> n; int x; for(int i=0;i<n;i++){ cin >> x; zabawki.push_back(x); } long long sum=0; bool used=0; bool ended=0; for(int x=0;x<n-1;x++){ // if(zabawki[x]==0 && zabawki[x+1]==0) // continue; if(zabawki[x]<0 || (zabawki[x]>0 && ended)){ sum = 1; break; } if(zabawki[x]>=zabawki[x+1] && !used && zabawki[x]!=0){ used=1; zabawki[x]--; } zabawki[x+1]-=zabawki[x];zabawki[x]-=zabawki[x]; if(used && !ended){ if(zabawki[x+1]==0){ ended=1; }else{ zabawki[x+1]--; } } } if(zabawki[n-1]<0)sum=1; if(zabawki[n-1]>1)sum=1; if(zabawki[n-1]==1 && used)sum=1; if(sum==0){ cout << "TAK\n"; }else{ cout << "NIE\n"; } } return 0; } |