zoukankan      html  css  js  c++  java
  • 【后缀自动机】hihocoder1445 后缀自动机二·重复旋律5

    #include<cstdio>
    #include<cstring>
    #include<iostream>
    using namespace std;
    typedef long long ll;
    #define MAXL 1000000
    #define MAXC 26
    char s[MAXL+10];//文本串
    int len/*文本串长度*/;
    struct SAM{
    	int n/*状态数0~n-1*/,maxlen[2*MAXL+10],minlen[2*MAXL+10],trans[2*MAXL+10][MAXC],slink[2*MAXL+10];
    	int new_state(int _maxlen,int _minlen,int _trans[],int _slink){
    		maxlen[n]=_maxlen;
    		minlen[n]=_minlen;
    		for(int i=0;i<MAXC;++i){
    			if(_trans==NULL){
    				trans[n][i]=-1;
    			}
    			else{
    				trans[n][i]=_trans[i];
    			}    
    		}
    		slink[n]=_slink;
    		return n++;
    	}
    	int add_char(char ch,int u,int pos){
    		if(u==-1){
    			return new_state(0,0,NULL,-1);
    		}
    		int c=ch-'a';
    		int z=new_state(maxlen[u]+1,-1,NULL,-1);
    		int v=u;
    		while(v!=-1 && trans[v][c]==-1){
    			trans[v][c]=z;
    			v=slink[v];
    		}
    		if(v==-1){//最简单的情况,suffix-path(u->S)上都没有对应字符ch的转移
    			minlen[z]=1;
    			slink[z]=0;
    			return z;
    		}
    		int x=trans[v][c];
    		if(maxlen[v]+1==maxlen[x]){//较简单的情况,不用拆分x
    			minlen[z]=maxlen[x]+1;
    			slink[z]=x;
    			return z;
    		}
    		int y=new_state(maxlen[v]+1,-1,trans[x],slink[x]);//最复杂的情况,拆分x
    		slink[y]=slink[x];
    		minlen[x]=maxlen[y]+1;
    		slink[x]=y;
    		minlen[z]=maxlen[y]+1;
    		slink[z]=y;
    		int w=v;
    		while(w!=-1 && trans[w][c]==x){
    			trans[w][c]=y;
    			w=slink[w];
    		}
    		minlen[y]=maxlen[slink[y]]+1;
    		return z;
    	}
    }sam;
    int m;
    ll ans;
    int main(){
    //	freopen("hihocoder1445.in","r",stdin);
    //	freopen("hihocoder1445.out","w",stdout);
    	char T[MAXL+10];
    	scanf("%s",s);
    	len=strlen(s);
    	int U=sam.add_char(0,-1,0);
    	for(int i=0;i<len;++i){
    		U=sam.add_char(s[i],U,i);
    	}
    	for(int i=1;i<sam.n;++i){
    		ans+=(ll)(sam.maxlen[i]-sam.minlen[i]+1);
    	}
    	cout<<ans<<endl;
    	return 0;
    }
  • 相关阅读:
    【vijos】【优先队列】合并果子
    【vijos】【二叉树】FBI树
    【NOIp复习】数据结构之栈、队列和二叉树
    【Leetcode】53. Maximum Subarray
    PHP json_encode转换空数组为对象
    206. Reverse Linked List
    151. Reverse Words in a String
    74. Search a 2D Matrix
    557. Reverse Words in a String III
    【Leetcode】79. Word Search
  • 原文地址:https://www.cnblogs.com/autsky-jadek/p/6636979.html
Copyright © 2011-2022 走看看