#pragma GCC optimize("O3") #include <bits/stdc++.h> using namespace std; #define ll long long #define pb push_back #define ff first #define ss second #define MOD 1000000009 #define INF 1000000019 #define INFL 1000000000000000099LL ll n,m,s,a,b,t; vector<ll>unq(vector<ll>v){ vector<ll>ans; sort(v.begin(),v.end()); reverse(v.begin(),v.end()); while(v.size() && v.back()<0)v.pop_back(); reverse(v.begin(),v.end()); if(v.empty())return {}; ans.pb(v[0]); for(ll i=1;i<v.size();i++){ if(v[i]!=v[i-1]) ans.pb(v[i]); } return ans; } vector<ll>dp[1000007][3];//zakladajac ze jak od prawej to juz jestem int main() { ios_base::sync_with_stdio(0);cin.tie(0); cin>>t; vector<ll>v; while(t--){ cin>>n; ll N=n; v.clear(); for(ll i=0;i<N;i++){ cin>>a; n--; if(a==0 && v.empty())continue; n++; v.pb(a); } while(v.size() && v.back()==0){ v.pop_back(); n--;} dp[0][0]={v[0]-1}; dp[0][1]={v[0]-1}; dp[0][2]={v[0]-1}; for(ll i=1;i<n;i++){ dp[i][0].clear(); dp[i][1].clear(); dp[i][2].clear(); for(ll j : dp[i-1][2]){ if(j>=1) dp[i][2].pb(v[i]-j); } for(ll j : dp[i-1][1]){ dp[i][2].pb(v[i]-1-j); } for(ll j : dp[i-1][0]){ dp[i][2].pb(v[i]-2-j); } dp[i][2]=unq(dp[i][2]); for( ll j : dp[i-1][0]){ dp[i][0].pb(v[i]-2-j); } dp[i][0]=unq(dp[i][0]); for( ll j : dp[i-1][0]){ dp[i][1].pb(v[i]-2-j); } for( ll j : dp[i-1][1]){ dp[i][1].pb(v[i]-1-j); } dp[i][1]=unq(dp[i][1]); } if(dp[n-1][2].size() && dp[n-1][2][0]==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 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 | #pragma GCC optimize("O3") #include <bits/stdc++.h> using namespace std; #define ll long long #define pb push_back #define ff first #define ss second #define MOD 1000000009 #define INF 1000000019 #define INFL 1000000000000000099LL ll n,m,s,a,b,t; vector<ll>unq(vector<ll>v){ vector<ll>ans; sort(v.begin(),v.end()); reverse(v.begin(),v.end()); while(v.size() && v.back()<0)v.pop_back(); reverse(v.begin(),v.end()); if(v.empty())return {}; ans.pb(v[0]); for(ll i=1;i<v.size();i++){ if(v[i]!=v[i-1]) ans.pb(v[i]); } return ans; } vector<ll>dp[1000007][3];//zakladajac ze jak od prawej to juz jestem int main() { ios_base::sync_with_stdio(0);cin.tie(0); cin>>t; vector<ll>v; while(t--){ cin>>n; ll N=n; v.clear(); for(ll i=0;i<N;i++){ cin>>a; n--; if(a==0 && v.empty())continue; n++; v.pb(a); } while(v.size() && v.back()==0){ v.pop_back(); n--;} dp[0][0]={v[0]-1}; dp[0][1]={v[0]-1}; dp[0][2]={v[0]-1}; for(ll i=1;i<n;i++){ dp[i][0].clear(); dp[i][1].clear(); dp[i][2].clear(); for(ll j : dp[i-1][2]){ if(j>=1) dp[i][2].pb(v[i]-j); } for(ll j : dp[i-1][1]){ dp[i][2].pb(v[i]-1-j); } for(ll j : dp[i-1][0]){ dp[i][2].pb(v[i]-2-j); } dp[i][2]=unq(dp[i][2]); for( ll j : dp[i-1][0]){ dp[i][0].pb(v[i]-2-j); } dp[i][0]=unq(dp[i][0]); for( ll j : dp[i-1][0]){ dp[i][1].pb(v[i]-2-j); } for( ll j : dp[i-1][1]){ dp[i][1].pb(v[i]-1-j); } dp[i][1]=unq(dp[i][1]); } if(dp[n-1][2].size() && dp[n-1][2][0]==0) cout<<"TAK\n"; else cout<<"NIE\n"; } return 0; } |