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[100010];

string KMPsolve(string s, string p) {
    nex[0] = 0;
    for(int i = 1, j = 0; i < p.size(); i++) {
        while(j && p[i] != p[j]) {
            j = nex[j - 1];
        }
        if(p[i] == p[j]) j++;
        nex[i] = j;
    }
    int j = 0;
    for(int i = 0; i < s.size(); i++) {
        while(j && s[i] != p[j]) {
            j = nex[j - 1];
        }
        if(s[i] == p[j]) j++;
        if(j == p.size() && (i + 1) != s.size()) {
            j = nex[j - 1];
        }
    }
    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 0;
}