#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,n) for (int i=(a);i<(n);i++)
#define per(i,a,n) for (int i=(n)-1;i>=(a);i--)
#define pb push_back
#define eb emplace_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
typedef vector<int> VI;
typedef basic_string<int> BI;
typedef long long ll;
typedef pair<int,int> PII;
typedef double db;
mt19937 mrand(1111);
const ll mod=1000000007;
int rnd(int x) { return mrand() % x;}
ll powmod(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;}
ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;}
// head
const int N=1010000;
vector<array<int,4>> dp,pd;
int n,a[N];
bool gao(VI a) {
dp.clear();
dp.pb({0,0,0,0});
rep(i,0,SZ(a)) {
pd.clear();
for (auto [vs,vt,l,r]:dp) {
rep(cs,0,2-vs) rep(ct,0,2-vt) {
int nl=a[i]-r-cs;
int nr=cs-ct-(l-r-nl);
if (nl<0||nr<0) continue;
if (i!=SZ(a)-1&&nl==0&&nr==0) continue;
pd.pb({vs+cs,vt+ct,nl,nr});
}
}
sort(all(pd)); pd.erase(unique(all(pd)),pd.end());
dp=pd;
}
return find(all(dp),array<int,4>{1,1,0,0})!=dp.end();
}
bool solve() {
scanf("%d",&n);
rep(i,1,n+1) scanf("%d",&a[i]);
int l=1,r=n;
while (a[l]==0) l++;
while (a[r]==0) r--;
rep(i,l,r+1) if (a[i]==0) return false;
return gao(VI(a+l,a+r+1));
}
int _;
int main() {
for (scanf("%d",&_);_;_--) {
puts(solve()?"TAK":"NIE");
}
}
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 | #include <bits/stdc++.h> using namespace std; #define rep(i,a,n) for (int i=(a);i<(n);i++) #define per(i,a,n) for (int i=(n)-1;i>=(a);i--) #define pb push_back #define eb emplace_back #define mp make_pair #define all(x) (x).begin(),(x).end() #define fi first #define se second #define SZ(x) ((int)(x).size()) typedef vector<int> VI; typedef basic_string<int> BI; typedef long long ll; typedef pair<int,int> PII; typedef double db; mt19937 mrand(1111); const ll mod=1000000007; int rnd(int x) { return mrand() % x;} ll powmod(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;} ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;} // head const int N=1010000; vector<array<int,4>> dp,pd; int n,a[N]; bool gao(VI a) { dp.clear(); dp.pb({0,0,0,0}); rep(i,0,SZ(a)) { pd.clear(); for (auto [vs,vt,l,r]:dp) { rep(cs,0,2-vs) rep(ct,0,2-vt) { int nl=a[i]-r-cs; int nr=cs-ct-(l-r-nl); if (nl<0||nr<0) continue; if (i!=SZ(a)-1&&nl==0&&nr==0) continue; pd.pb({vs+cs,vt+ct,nl,nr}); } } sort(all(pd)); pd.erase(unique(all(pd)),pd.end()); dp=pd; } return find(all(dp),array<int,4>{1,1,0,0})!=dp.end(); } bool solve() { scanf("%d",&n); rep(i,1,n+1) scanf("%d",&a[i]); int l=1,r=n; while (a[l]==0) l++; while (a[r]==0) r--; rep(i,l,r+1) if (a[i]==0) return false; return gao(VI(a+l,a+r+1)); } int _; int main() { for (scanf("%d",&_);_;_--) { puts(solve()?"TAK":"NIE"); } } |
English