codeforces 432D 【kmp+dp】||【后缀数组】

codeforces 432D 【kmp+dp】||【后缀数组】
Description:
给出一个字符串,求有多少种长度的前缀和后缀相等,并且得到的这个前缀在原字符串中出现的次数。

 solution1:kmp+dp
 求前缀出现的次数,原来就写过:
 https://blog.csdn.net/qq_43653111/article/details/103354633
 现在找一下前缀后缀相等:很容易发现n,nxt[n]就是一个
 再找找nxt[nxt[n]]也是,gg了

    scanf("%s",s);
    len=strlen(s);
    kmp();
    fill(dp,dp+len+1,1);
    for(int i=len; i>=1; i--) {
    
    
        dp[nxt[i]]+=dp[i];
    }
    int now=len;
    while(~nxt[now]) {
    
    
        ans.push_back(now);
        now=nxt[now];
    }
    reverse(ans.begin(),ans.end());
    printf("%d\n",ans.size());
    for(auto v:ans)
    {
    
    
        printf("%d %d\n",v,dp[v]);
    }
solution2:后缀数组
这题很难想到后缀数组;后缀数组一般是处理子串问题的,
首先我们看整个串的前缀 和 其他位置lcp如果sa[i]+mi==n+1
不就说明前缀后缀相等了吗?有点像那个exkmp
具体做法:我们从下标为1的rank前后扫一遍,满足条件的就是
前后缀相等的,标记一下。
怎么求前缀出现次数?
我们初始化cnt[n]=1,那么它的前缀部分min[height[i]]肯定
也是其中的,长串会对短的有贡献,
我们后缀和加一下就行了,类似于那个kmp+dp

    scanf("%s",a+1);
    n=strlen(a+1);
    SA.init();
    int st=rk[p][1],mi=INF;
    for(int i=st+1; i<=n; i++) {
    
    
        mi=min(mi,height[i]);

        if(mi+sa[p][i]==n+1) {
    
    
            vis[mi]=1;
        }
        cnt[mi]++;
    }
    mi=INF;
    for(int i=st-1; i>=1; i--) {
    
    
        mi=min(mi,height[i+1]);
        if(mi+sa[p][i]==n+1) {
    
    
            vis[mi]=1;
        }
        cnt[mi]++;
    }
    vis[n]=cnt[n]=1;
    for(int i=n;i>=1;i--)
        cnt[i]+=cnt[i+1];
    printf("%d\n",accumulate(vis+1,vis+n+1,0));
    for(int i=1;i<=n;i++)
    {
    
    
        if(vis[i])
            printf("%d %d\n",i,cnt[i]);
    }

猜你喜欢

转载自blog.csdn.net/qq_43653111/article/details/106669431