Source code for submission s648

bugs.cpp

  1. #include <algorithm>
  2. #include <cctype>
  3. #include <cmath>
  4. #include <complex>
  5. #include <cstdio>
  6. #include <cstring>
  7. #include <iomanip>
  8. #include <iostream>
  9. #include <list>
  10. #include <map>
  11. #include <queue>
  12. #include <set>
  13. #include <sstream>
  14. #include <stack>
  15. #include <string>
  16. #include <utility>
  17. #include <vector>
  18. using namespace std;
  19.  
  20. #define DEBUG(x) cout << ">>> " << #x << " : " << x << endl;
  21. #define REP(i,a) for (int i = 0; i < (a); ++i)
  22. #define FOR(i,a,b) for (int i = (a); i <= (b); ++i)
  23. #define FORD(i,a,b) for (int i = (a); i >= (b); --i)
  24. inline bool EQ(double a, double b) { return fabs(a-b) < 1e-9; }
  25.  
  26. const int INF = 1<<29;
  27. typedef long long ll;
  28. ///////////////////////////////////////////////////////////////////////////
  29.  
  30. const int MAX = 1007, MAX2 = 2000007;
  31. int T, shift[MAX], state[MAX2];
  32. char bug[MAX], input[MAX2];
  33.  
  34. int main()
  35. {
  36. while (scanf("%d %s\n", &T, bug) == 2)
  37. {
  38. int len = strlen(bug);
  39. shift[0] = -1;
  40. for (int i = 0, j = -1; i < len; shift[++i] = ++j)
  41. while (j >= 0 && bug[i] != bug[j])
  42. j = shift[j];
  43.  
  44. while (T--)
  45. {
  46. gets(input);
  47. int len2 = 0, j = 0;
  48. for (int i = 0; input[i]; ++i)
  49. {
  50. input[len2] = input[i];
  51. while (j >= 0 && bug[j] != input[i])
  52. j = shift[j];
  53. ++j;
  54. state[len2] = j;
  55. ++len2;
  56. if (j == len)
  57. {
  58. len2 -= len;
  59. j = (len2?state[len2-1]:0);
  60. }
  61. }
  62. input[len2] = 0;
  63. printf("%s\n", input);
  64. }
  65. }
  66.  
  67. return 0;
  68. }
  69.