#include <cstdio>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <set>
using namespace std;
typedef vector<int> VI;
typedef vector<VI> VVI;
typedef pair<int, int> PI;
typedef long long LL;
#define FOR(x, b, e) for(int x = b; x <= (e); ++x)
#define FORD(x, b, e) for(int x = b; x >= (e); --x)
#define REP(x, n) for(int x = 0; x < (n); ++x)
#define VAR(v, n) __typeof(n) v = (n)
#define ALL(c) (c).begin(), (c).end()
#define SIZE(x) ((int)(x).size())
#define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i)
#define PB push_back
#define MP make_pair
#define ST first
#define ND second
const int INF = 1000000001;
const double EPS = 10e-9;
void Fibonacci(set<int> &k)
{
VI tab;
tab.PB(0);
tab.PB(1);
while(tab.back() < INF) tab.PB(tab[SIZE(tab) - 1] + tab[SIZE(tab) - 2]);
REP(x, SIZE(tab)) REP(y, SIZE(tab)) k.insert(tab[x] * tab[y]);
}
int main()
{
int n, b;
set<int> k;
scanf("%d", &n);
Fibonacci(k);
REP(x, n)
{
scanf("%d", &b);
if(k.find(b) == k.end()) printf("NIE\n");
else printf("TAK\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 | #include <cstdio> #include <iostream> #include <algorithm> #include <string> #include <vector> #include <set> using namespace std; typedef vector<int> VI; typedef vector<VI> VVI; typedef pair<int, int> PI; typedef long long LL; #define FOR(x, b, e) for(int x = b; x <= (e); ++x) #define FORD(x, b, e) for(int x = b; x >= (e); --x) #define REP(x, n) for(int x = 0; x < (n); ++x) #define VAR(v, n) __typeof(n) v = (n) #define ALL(c) (c).begin(), (c).end() #define SIZE(x) ((int)(x).size()) #define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define PB push_back #define MP make_pair #define ST first #define ND second const int INF = 1000000001; const double EPS = 10e-9; void Fibonacci(set<int> &k) { VI tab; tab.PB(0); tab.PB(1); while(tab.back() < INF) tab.PB(tab[SIZE(tab) - 1] + tab[SIZE(tab) - 2]); REP(x, SIZE(tab)) REP(y, SIZE(tab)) k.insert(tab[x] * tab[y]); } int main() { int n, b; set<int> k; scanf("%d", &n); Fibonacci(k); REP(x, n) { scanf("%d", &b); if(k.find(b) == k.end()) printf("NIE\n"); else printf("TAK\n"); } return 0; } |
polski