Advertisement
lalalalalalalaalalla

Untitled

Sep 22nd, 2019
190
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 4.61 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3. #include <string>
  4. #include <iomanip>
  5. #include <queue>
  6. #include <cmath>
  7. #include <algorithm>
  8. #include <tuple>
  9. #include <iomanip>
  10. #include <stdio.h>
  11. #include <numeric>
  12. #include <map>
  13. #include <bitset>
  14. #include <set>
  15. #include <stack>
  16. #include <queue>
  17.  
  18. /*
  19. #pragma GCC optimize("Ofast,no-stack-protector")
  20. #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,avx2,tune=native")
  21. #pragma GCC optimize("unroll-loops")
  22. #pragma GCC optimize("fast-math")
  23. #pragma GCC optimize("section-anchors")
  24. #pragma GCC optimize("profile-values,profile-reorder-functions,tracer")
  25. #pragma GCC optimize("vpt")
  26. #pragma GCC optimize("rename-registers")
  27. #pragma GCC optimize("move-loop-invariants")
  28. #pragma GCC optimize("unswitch-loops")
  29. #pragma GCC optimize("function-sections")
  30. #pragma GCC optimize("data-sections")
  31. #pragma GCC optimize("branch-target-load-optimize")
  32. #pragma GCC optimize("branch-target-load-optimize2")
  33. #pragma GCC optimize("btr-bb-exclusive")
  34. */
  35.  
  36. #define int long long
  37. #define ll long long
  38. #define ull unsigned long long
  39. #define all(a) a.begin(), a.end()
  40. #define pii pair<int, int>
  41. #define pb push_back
  42. #define ld long double
  43.  
  44.  
  45. using namespace std;
  46.  
  47. const int INF = 1e13;
  48. const int mod = 2600000069;
  49. const int p = 179;
  50.  
  51. string s;
  52. int n, l, r, mid;
  53. vector<vector<int>> ah, bh;
  54. vector<set<int>> sla, slb;
  55. vector<int> pw(100001);
  56.  
  57. inline int subh(vector<int>& h, int l, int r) {
  58. int big = h[r], small = 0;
  59. if (l > 0) {
  60. small = (h[l - 1] * pw[r - l + 1]) % mod;
  61. }
  62. return (big - small + mod) % mod;
  63. }
  64.  
  65. signed main() {
  66. ios_base::sync_with_stdio(0);
  67. cin.tie(0);
  68. cout.tie(0);
  69. // freopen("input.in", "r", stdin);
  70. // freopen("output.out", "w", stdout);
  71. pw[0] = 1;
  72. for (int i = 1; i < 100001; i++) {
  73. pw[i] = (pw[i - 1] * p) % mod;
  74. }
  75. int n, m;
  76. cin >> n >> m;
  77. ah.resize(n);
  78. bh.resize(m);
  79. sla.resize(n);
  80. slb.resize(m);
  81. vector<string> a(n), b(m);
  82. int q;
  83. for (int i = 0; i < n; i++) {
  84. cin >> a[i];
  85. q = a[i].size();
  86. ah[i].resize(q);
  87. ah[i][0] = a[i][0] - 'a' + 1;
  88. for (int j = 1; j < q; j++) {
  89. ah[i][j] = ((ah[i][j - 1] * p) % mod + a[i][j] - 'a' + 1) % mod;
  90. }
  91. if (q == 1) {
  92. sla[i].insert(0);
  93. sla[i].insert(ah[i][0]);
  94. continue;
  95. }
  96. // cout << q << "\n";
  97. // cout << a[i] << "\n";
  98. // cout << ah[i].size() << "\n";
  99. sla[i].insert(ah[i][q - 1]);
  100. // cout << "1: " << ah[i][q - 1] << "\n";
  101. sla[i].insert(ah[i][q - 2]);
  102. // cout << "2: " << ah[i][q - 2] << "\n";
  103. sla[i].insert(subh(ah[i], 1, q - 1));
  104. // cout << "3: "<< subh(ah[i], 1, q - 1) << "\n";
  105. for (int j = 1; j < q - 1; j++) {
  106. // cout << "here\n";
  107. // cout << ah[i].size() << " " << 0 << " " << i - 1 << "\n" << i + 1 << " " << q - 1 << "\n";
  108. sla[i].insert((subh(ah[i], j + 1, q - 1) + (subh(ah[i], 0, j - 1) * pw[q-1-j]) % mod) % mod);
  109. }
  110. }
  111. for (int i = 0; i < m; i++) {
  112. cin >> b[i];
  113. q = b[i].size();
  114. bh[i].resize(q);
  115. bh[i][0] = b[i][0] - 'a' + 1;
  116. for (int j = 1; j < q; j++) {
  117. bh[i][j] = ((bh[i][j - 1] * p) % mod + b[i][j] - 'a' + 1) % mod;
  118. }
  119. if (q == 1) {
  120. slb[i].insert(0);
  121. slb[i].insert(bh[i][0]);
  122. } else {
  123. slb[i].insert(bh[i][q - 1]);
  124. slb[i].insert(bh[i][q - 2]);
  125. slb[i].insert(subh(bh[i], 1, q - 1));
  126. for (int j = 1; j < q - 1; j++) {
  127. slb[i].insert((subh(bh[i], j + 1, q - 1) + (subh(bh[i], 0, j - 1) * pw[q-1-j]) % mod) % mod);
  128. }}
  129. int cnt = 0;
  130. string ans;
  131. for (int j = 0; j < n; j++) {
  132. set<int> k;
  133. set_intersection(all(sla[j]), all(slb[i]), inserter(k, k.begin()));
  134. // cout << k.size() << "\n";
  135. // if (k.size() == 0) {
  136. // for (auto q : sla[j]) cout << q << " ";
  137. // cout << "\n";
  138. // for (auto q : slb[i]) cout << q << " ";
  139. // cout << "\n";
  140. // }
  141. if (k.size() > 0) {
  142. cnt++;
  143. ans = a[j];
  144. }
  145. }
  146. if (cnt != n) {
  147. cout << "\nBAD\n";
  148. // return 1;
  149. }
  150. cout << "OK\n";
  151. // return 0;
  152. // if (cnt == 1) {
  153. // cout << b[i] << " 1 " << ans << "\n";
  154. // } else {
  155. // cout << b[i] << " " << cnt << "\n";
  156. // }
  157. }
  158.  
  159. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement