HDU/HDOJ 1867 A + B for you again

仔细了解KMP之后再看这题就会发现是裸题。

因为kmp我们可以求出s的f数组，表示能与p的多少前缀匹配。那么我们只需取f[s.size() - 1]即可。

 #include <cstdio>
 #include <string>
 #include <iostream>
 using std::string;
 using std::cin;
 using std::cout;
 using std::endl;

 int nex[];

 string KMPsolve(string s, string p) {
     nex[] = ;
     for(int i = , j = ; i < p.size(); i++) {
         while(j && p[i] != p[j]) {
             j = nex[j - ];
         }
         if(p[i] == p[j]) j++;
         nex[i] = j;
     }
     int j = ;
     for(int i = ; i < s.size(); i++) {
         while(j && s[i] != p[j]) {
             j = nex[j - ];
         }
         if(s[i] == p[j]) j++;
         if(j == p.size() && (i + ) != s.size()) {
             j = nex[j - ];
         }
     }
     for(; j < p.size(); j++) {
         s += p[j];
     }
     return s;
 }

 int main() {
     string s, p;
     while(std::cin >> s >> p) {
     string a = KMPsolve(s, p);
     string b = KMPsolve(p, s);
     if(a.size() < b.size()) {
         cout << a << endl;
     }
     else if(a.size() > b.size()) {
         cout << b << endl;
     }
     else {
         cout << (a < b ? a : b) << endl;
     }
     }
     return ;
 }

AC代码