KMP (Knuth-Morris-Pratt) algorithm introduction to algorithm 3rd, example 32.4

算法导论 第三版 32.4 KMP

//KMP (Knuth-Morris-Pratt) algorithm introduction to algorithm 3rd, example 32.4


#include <iostream>
#include <vector>
#include <algorithm>
#include <cassert>
#include <ostream>
#include <iterator>
#include <limits>
#include <complex>
#include "stopwatch.h"


std::vector<size_t> compute_prefix(const std::string & str)
{
    std::vector<size_t> prefix(str.size(), 0);
    size_t ip = 0;
    for (size_t is = 1; is < str.size(); ++is)
    {
        while (ip > 0 && str[ip] != str[is])
        {
            ip = prefix[ip-1];
        }
        if (str[ip] == str[is])
        {
            ++ip;
        }
        prefix[is] = ip;
    }

    return prefix;
}

std::vector<size_t> kmp_match(const std::string &s1, const std::string & s2, const std::vector<size_t> & prefixes)
{
    std::vector<size_t> indexes;
    size_t ip = 0;
    for (size_t is = 0; is < s1.size(); ++is)
    {
        while (ip > 0 && s1[is] != s2[ip])
        {
            ip = prefixes[ip-1];
        }

        if (s1[is] == s2[ip])
        {
            ++ip;
        }
        if (ip == prefixes.size())
        {
            ip = prefixes[ip-1];
            indexes.push_back(is);
        }
    }

    return indexes;
}

void test(std::string && s1, std::string && s2)
{
    std::cout << "Match " << s1 << "  with " << s2 << std::endl;
    Stopwatch watch;
    watch.start();
    auto prefixes = compute_prefix(s2);
    auto indexes = kmp_match(s1, s2, prefixes);
    watch.stop();
    std::cout << "prefix of " << s2 << " is ";
    std::copy(prefixes.begin(), prefixes.end(), std::ostream_iterator<size_t>(std::cout, " "));
    std::cout << std::endl;
    std::cout << "Match string at index ";
    std::copy(indexes.begin(), indexes.end(), std::ostream_iterator<size_t>(std::cout, " "));
    std::cout << " use time " << watch.elapsed() << " ms\n" << std::endl;
}

int main()
{
    std::cout << "start" << std::endl;
    test(std::string("aaababaabaababaab"), std::string("aabab"));
    test(std::string("bacbababacaabcbab"), std::string("ababaca"));
    test(std::string("aaabaabaabaabaabaab"), std::string("aabaab"));

    return 0;
}