#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; } |