fork(1) download
  1. #include <iostream>
  2. #include <vector>
  3. #include <string>
  4. #include <cmath>
  5.  
  6. using namespace std;
  7.  
  8. int main()
  9. {
  10. ios_base::sync_with_stdio(0);
  11. int t,k,d,w,n;
  12. double q;
  13. vector <int> tab;
  14. string s,tmp;
  15.  
  16. cin>>t;
  17. while(t--)
  18. {
  19. s.clear();
  20. tmp.clear();
  21. cin>>s;
  22. k=s.size();
  23. for(int i=0; i<k; i++)
  24. {
  25. if(s[i]=='A')
  26. {
  27. n=i;
  28. break;
  29. }
  30. }
  31. if(n>0)
  32. {
  33. for(int i=0; i<n; i++) {tmp+=s[i];}
  34. s.erase(0,n);
  35. s.append(tmp);
  36. }
  37.  
  38. q=sqrt((double)k);
  39. for(int i=1; i<=q; i++)
  40. {
  41. if(k%i==0)
  42. {
  43. if(k/i<k/2) tab.push_back(k/i);
  44. tab.push_back(i);
  45. }
  46. }
  47.  
  48. for(int i=0; i<tab.size(); i++)
  49. {
  50. d=tab[i]; w=1;
  51. for(int j=0; j<k; j+=d)
  52. {
  53. if(s[j]!='A') {w=0; break;}
  54. }
  55. if(w) break;
  56. }
  57. if(w) cout<<"TAK"<<endl;
  58. else cout<<"NIE"<<endl;
  59. tab.clear();
  60. }
  61. return 0;
  62. }
  63.  
Success #stdin #stdout 0s 3456KB
stdin
4
AAA
AABA
ABABA
ABAAAB
stdout
TAK
NIE
NIE
TAK