#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
class hash_class
{
const vector<uint32_t> primeList = {50131, 45541, 51421};
const uint32_t p = 31;
vector<uint32_t> p_pow;
public:
vector<uint32_t> h_list; // lista hashy dla slowa czytanego od poczatku
vector<uint32_t> hr_list; // lista hashy dla slowa czytanego od konca
hash_class()
{
for (uint32_t ii=0; ii<primeList.size(); ii++)
{
p_pow.push_back(1);
h_list.push_back(0);
hr_list.push_back(0);
}
}
void update_hash(char val)
{
for (uint32_t ii=0; ii<h_list.size(); ii++)
{
h_list[ii] = (h_list[ii]*p +val) %primeList[ii];
hr_list[ii] += (val*p_pow[ii])%primeList[ii];
hr_list[ii] %= primeList[ii];
p_pow[ii] = (p_pow[ii]*p) %primeList[ii];
}
}
bool compare_hash()
{
for (uint32_t ii=0; ii<h_list.size(); ii++)
{
if (h_list[ii] != hr_list[ii])
return false;
}
return true;
}
};
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
uint32_t N;
cin>>N;
hash_class h;
char c;
if (N>0)
for (uint32_t ii=0; ii<N; ii++)
{
cin>>c;
h.update_hash(c);
}
else while (cin>>c && isalpha(c))
{
h.update_hash(c);
}
if (h.compare_hash()==true)
cout<<"TAK"<<endl;
else
cout<<"NIE"<<endl;
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 | #include <iostream> #include <vector> #include <algorithm> using namespace std; class hash_class { const vector<uint32_t> primeList = {50131, 45541, 51421}; const uint32_t p = 31; vector<uint32_t> p_pow; public: vector<uint32_t> h_list; // lista hashy dla slowa czytanego od poczatku vector<uint32_t> hr_list; // lista hashy dla slowa czytanego od konca hash_class() { for (uint32_t ii=0; ii<primeList.size(); ii++) { p_pow.push_back(1); h_list.push_back(0); hr_list.push_back(0); } } void update_hash(char val) { for (uint32_t ii=0; ii<h_list.size(); ii++) { h_list[ii] = (h_list[ii]*p +val) %primeList[ii]; hr_list[ii] += (val*p_pow[ii])%primeList[ii]; hr_list[ii] %= primeList[ii]; p_pow[ii] = (p_pow[ii]*p) %primeList[ii]; } } bool compare_hash() { for (uint32_t ii=0; ii<h_list.size(); ii++) { if (h_list[ii] != hr_list[ii]) return false; } return true; } }; int main() { ios_base::sync_with_stdio(0); cin.tie(0); uint32_t N; cin>>N; hash_class h; char c; if (N>0) for (uint32_t ii=0; ii<N; ii++) { cin>>c; h.update_hash(c); } else while (cin>>c && isalpha(c)) { h.update_hash(c); } if (h.compare_hash()==true) cout<<"TAK"<<endl; else cout<<"NIE"<<endl; return 0; } |
English