傅芳杰

个人博客

You Deserve The Best


KMP

AcWing 831. KMP字符串 原题链接

给定一个模式串S,以及一个模板串P,所有字符串中只包含大小写英文字母以及阿拉伯数字。

模板串P在模式串S中多次作为子串出现。

求出模板串P在模式串S中所有出现的位置的起始下标。

输入格式

第一行输入整数N,表示字符串P的长度。

第二行输入字符串P。

第三行输入整数M,表示字符串S的长度。

第四行输入字符串S。

输出格式

共一行,输出所有出现位置的起始下标(下标从0开始计数),整数之间用空格隔开。

数据范围

1≤N≤1051≤N≤105 1≤M≤1061≤M≤106

输入样例:

3
aba
5
ababa

输出样例:

0 2
public static void main(String[] args) {
        int n = in.nextInt();
        String pattern = in.next();
        char[] cp = new char[n + 1];
        for (int i = 1; i < cp.length; i++) {
            cp[i] = pattern.charAt(i - 1);
        }
        int m = in.nextInt();
        String str = in.next();
        char[] cs = new char[m + 1];
        for (int i = 1; i < cs.length; i++) {
            cs[i] = str.charAt(i - 1);
        }
        int[] nextArray = getNextArray(cp);
        for (int i = 1, j = 0; i < cs.length; i++) {
            while (j != 0 && cs[i] != cp[j + 1]) {
                j = nextArray[j];
            }
            if (cs[i] == cp[j + 1]) {
                j++;
            }
            if (j == n) {
                out.print(i - n + " ");
                j = nextArray[j];
            }
        }
        out.flush();
        out.close();
    }
    private static int[] getNextArray(char[] cs) {
        int[] next = new int[cs.length];
        for (int i = 2, j = 0; i < cs.length; i++) {
            while (j != 0 && cs[i] != cs[j + 1]) {
                j = next[j];
            }
            if (cs[i] == cs[j + 1]) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }