Codeforces 432D Prefixes and Suffixes：KMP + dp

題目連結：http://codeforces.com/problemset/problem/432/D

題意：

　　給你一個字串s，讓你找出所有既是字首又是字尾的子串，並輸出它們分別出現了多少次。

 

題解：

　　先對原串求一次nex陣列。

　　然後列舉位置i：

　　　　sub(k)表示字首s[0 to k]

　　　　dp[i]表示sub(i)在原串s中的出現次數

　　　　假設nex[i] = a, nex[a] = b, nex[b] = c ... nex[z] = -1

　　　　則sub(a), sub(b), sub(c)...sub(z)均以s[i]結尾，dp[a...z]均+1。

　　然而一個一個加會T...

　　

　　所以:

　　　　初始時所有的 dp = 1

　　　　每次：if(nex[i] != -1) dp[nex[i]] += dp[i]

　　　　一路傳遞下去就好。

　　

　　最後從nex[len-1]開始往前跳nex。

　　對於每次跳到的nex，sub(nex)一定也是s的字尾。

　　此時輸出它的出現次數dp[nex]。

　　另外因為要順著輸出，而nex是倒著跳的，所以先存到stack裡再輸出。

 

AC Code:

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stack>
#define MAX_N 100005

using namespace std;

int n;
int dp[MAX_N];
int nex[MAX_N];
char s[MAX_N];

void cal_nex(char *p,int len)
{
    nex[0]=-1;
    int k=-1;
    for(int i=1;i<len;i++)
    {
        while(k>-1 && p[i]!=p[k+1]) k=nex[k];
        if(p[i]==p[k+1]) k++;
        nex[i]=k;
    }
}

int main()
{
    scanf("%s",s);
    n=strlen(s);
    cal_nex(s,n);
    for(int i=0;i<n;i++) dp[i]=1;
    for(int i=n-1;i>=0;i--)
    {
        if(nex[i]!=-1) dp[nex[i]]+=dp[i];
    }
    stack<int> stk;
    int p=n-1;
    while(p!=-1)
    {
        stk.push(p);
        p=nex[p];
    }
    cout<<stk.size()<<endl;
    while(!stk.empty())
    {
        int now=stk.top();
        stk.pop();
        cout<<now+1<<" "<<dp[now]<<endl;
    }
}