码迷,mamicode.com
首页 > 编程语言 > 详细

POJ-2752(KMP算法+前缀数组的应用)

时间:2019-08-09 13:13:20      阅读:69      评论:0      收藏:0      [点我收藏+]

标签:ring   i++   using   算法   name   The   std   ble   一个   

Seek the Name, Seek the Fame

POJ-2752

  • 本题使用的算法还是KMP
  • 最主要的片段就是前缀数组pi的理解,这里要求解的纸盒pi[n-1]有关,但是还是需要使用一个循环来依次找到前面符合的前缀(所谓符合就是可以保持既是前缀也是s的后缀的子串长度)。
#include<iostream>
#include<cstdio>
#include<string>
#include<algorithm>
#include<set>
#include<cstring>
using namespace std;
const int N=400005;
int pi[N];
void Pi(string s){
    memset(pi,0,sizeof(pi));
    int n=s.length();
    pi[0]=0;
    for(int i=1;i<n;i++){
        int j=pi[i-1];
        while(j>0&&s[i]!=s[j]){
            j=pi[j-1];
        }
        if(s[i]==s[j])
            j++;
        pi[i]=j;
    }
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    string s;
    while(cin>>s){
        set<int> se;
        Pi(s);
        int n=s.length();
        if(n==1){
            cout<<1<<endl;
            continue;
        }
        int j=pi[n-2];
        while(j>0){
            if(s[n-1]==s[j]){
                se.insert(j+1);
            }
            j=pi[j-1];
        }
        se.insert(n);
        if(s[0]==s[n-1])
        se.insert(1);
        int k=0;
        int cnt=(int)se.size();
        for(set<int>::iterator it=se.begin();it!=se.end();it++){
            if(k++==cnt-1)
                cout<<*it<<endl;
            else
                cout<<*it<<" ";
        }
    }
    return 0;
}

POJ-2752(KMP算法+前缀数组的应用)

标签:ring   i++   using   算法   name   The   std   ble   一个   

原文地址:https://www.cnblogs.com/GarrettWale/p/11326449.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!