2016 计蒜之道初赛第三场百度帐号的选取方案（中等） KMP

本文链接：https://blog.csdn.net/danliwoo/article/details/51591137

原题见计蒜之道

题意描述如下：记一个字符串最大循环节的个数为p。一个长为N的字符串中取两个不相交的子串，使得两个子串的p相等，问有多少种取法。
三个档次的N分别为50，1000，100000.

思路

求字符串的循环节，利用KMP算法中的预处理数组next来处理。可自行先A了POJ 2406再继续做。

对于最小范围的N来说，O( $N^4$ )两两枚举子串后再求p值，复杂度O( $N^5$ )可过。但是N到了1000，则要优化重复KMP造成的浪费。

首先O( $N^2$ )得到所有子串的p值，用a[i][j]保存子串Str[i..j]的p值。
对于所有的Str[i…]，从i开始的各个子串均有自己的p值，存放在b[i][p]中。即b[i][j]表示的是从i开始，p值等于j的子串的个数。

扫描子串Str[i..j]作为左边的子串，则其右边的满足的子串个数就应该是 $\sum_{x=j+1}^{length}b[x][a[i][j]]$ 事实上，对于同一p值的b[i][p]可以做一个预处理，保存其后缀和，之后O(1)使用即可。

最后一个层次的还没做出来，会做的大神求题解啊。。

附代码

/*--------------------------------------------
 * File Name: 2016 计蒜之道 初赛 第三场 第三题
 * Author: Danliwoo
 * Mail: Danliwoo@outlook.com
 * Created Time: 2016-06-05 20:08:30
--------------------------------------------*/

#include <bits/stdc++.h>
using namespace std;
#define N 1005
#define LL long long
LL a[N][N], b[N][N], s[N][N];
int nxt[1000005];
void getnxt(string s){
    int i=0,j=-1;
    nxt[0]=-1;
    int len=s.length();
    while(i<len){
        if(s[i]==s[j]||j==-1){
            i++;
            j++;
            nxt[i]=j;
        }
        else
            j=nxt[j];
    }
}
int getnum(string s){
    int len=s.length();
    getnxt(s);
    if(len%(len-nxt[len])==0)
        return len/(len-nxt[len]);
    return 1;
}
string str;
int main(){
    while(cin >> str){
        int len = str.length();
        LL ans = 0;
        memset(a, 0, sizeof(a));
        memset(b, 0, sizeof(b));
        memset(s, 0, sizeof(s));
        for(int i = 0;i < len;i++)
            for(int j = i;j < len;j++){
                a[i][j] = getnum(str.substr(i, j-i+1)); 
                //这里可以优化一下，使得复杂度是真正的O(N^2)
                b[i][a[i][j]]++;
            }
        for(int i = len-1;i >= 0;i--){
            for(int j = 0;j < len;j++){
                if(i == len-1) s[i][j] = b[i][j];
                else s[i][j] = s[i+1][j] + b[i][j];
            }
        }
        for(int i = 0;i < len-1;i++)
            for(int j = i;j < len-1;j++){
                int p = a[i][j];
                ans += s[j+1][p];
            }
        printf("%lld\n", ans);
    }
}