kuangbin专题16H(next数组)

题目链接: https://vjudge.net/contest/70325#problem/H

 

题意: 输入字符串 str, 求 str 子串中既是 str 前缀又是 str 后缀的的字符串长度, 按照升序输出.

 

思路: 先求个 next 数组, next[i] 为以 i - 1(字符串下标从0开始) 字符为后缀的字符串与以第一个字符为前缀的字符串的最大匹配长度.

首先 str 本身是满足条件的. 然后再不断用 next[len] 迭代 len 即可. len 的初始值为 str 的长度. 由 next 的性质可以知道以 next[len] - 1 为后缀的字符串会和以 len - 1 为后缀的字符串重合, 那么所有迭代到的子串都会和以 str 最后一个字符为后缀的字符串重合. (可以参考下面的图) 那么现在前后缀条件都满足啦.

图是从 http://www.cnblogs.com/dongsheng/archive/2012/08/13/2636261.html 偷过来的....

代码:

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;

const int MAXN = 4e5 + 10;
int nxt[MAXN], vis[MAXN], len;
char str[MAXN];

void get_nxt(void){
    memset(nxt, 0, sizeof(nxt));
    int i = 0, j = -1;
    nxt[0] = -1;
    while(i < len){
        if(j == -1 || str[i] == str[j]) nxt[++i] = ++j;
        else j = nxt[j];
    }
}

int main(void){
    while(~scanf("%s", str)){
        len = strlen(str);
        int cnt = 0;
        get_nxt();
        vis[cnt++] = len;
        while(nxt[len] > 0){
            len = nxt[len];
            vis[cnt++] = len;
        }
        for(int i = cnt - 1; i >= 0; i--){
            printf("%d", vis[i]);
            if(i) printf(" ");
        }
        puts("");
    }
    return 0;
}