#include <cstdio>
#include <algorithm>
using namespace std;
int A[1000000];
int main()
{
int t, n;
scanf("%i ", &t);
while (t--)
{
scanf("%i ", &n);
for (int i = 0; i<n; i++) scanf("%i ", A+i);
for (int i = 0, l = 0; i<n-1; i++)
{
int m = min(A[i]-l, A[i+1]-1);
if (m>0) A[i] -= m, A[i+1] -=m;
if (A[i]>0) l = 1;
}
for (int i = n-1, l = 0; i>0; i--)
{
int m = min(A[i]-l, A[i-1]-1);
if (m>0) A[i] -= m, A[i-1] -=m;
if (A[i]>0) l = 1;
}
int p = 0, q = 0, f = 0;
for (int i = 0; i<n; i++)
{
if (A[i]!=p) p = A[i], q++;
if (A[i]>1 || q>2) {f = 1; break;}
}
puts(f? "NIE": "TAK");
}
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 | #include <cstdio> #include <algorithm> using namespace std; int A[1000000]; int main() { int t, n; scanf("%i ", &t); while (t--) { scanf("%i ", &n); for (int i = 0; i<n; i++) scanf("%i ", A+i); for (int i = 0, l = 0; i<n-1; i++) { int m = min(A[i]-l, A[i+1]-1); if (m>0) A[i] -= m, A[i+1] -=m; if (A[i]>0) l = 1; } for (int i = n-1, l = 0; i>0; i--) { int m = min(A[i]-l, A[i-1]-1); if (m>0) A[i] -= m, A[i-1] -=m; if (A[i]>0) l = 1; } int p = 0, q = 0, f = 0; for (int i = 0; i<n; i++) { if (A[i]!=p) p = A[i], q++; if (A[i]>1 || q>2) {f = 1; break;} } puts(f? "NIE": "TAK"); } return 0; } |
English