后缀自动机模板(保存两个,第二个是杭电多校标称看到的)

SAM常解决的问题:

  • 问题1.给定文本T,询问格式如下:给定字符串P,问P是否是T的子串。

  • 问题2.给定字符串S,问它有多少不同的子串。(模板中的GetSubNum函数)

  • 问题3.给定字符串S,求其所有不同子串的总长度。

  • 问题4.给定字符串S,一系列询问——给出整数K_i,计算S的所有子串排序后的第K_i个。

  • 问题5.给定字符串S,找到和它循环同构的字典序最小字符串。

  • 问题.给定文本T,询问格式如下:给定字符串P,希望找出P作为子串在文本T中出现了多少次(出现区间可以相交)。

  • 问题6.给定文本T,询问格式如下:给定字符串P,求P在文本中第一次出现的位置。

  • 问题7.给定文本T,询问格式如下:给定字符串P,要求给出P在T中的所有出现位置(出现区间可以相交)。

  • 问题8.给定两个字符串S和T。要求找出它们的最长公共子串,即一个字符串X,它同时是S和T的子串。

#include<bits/stdc++.h>
#include <time.h>
#define int long long
#define maxc 28
using namespace std;
const int maxn = 5e5+ 5;
const int mod = 1e9 + 7;
typedef long long ll;
int len[maxn * 2], //最长子串的长度(该节点字串数量=len[x]-len[link[x]])
link[maxn * 2],   //后缀链接(最短串前部减少一个字符所到达的状态)
cnt[maxn * 2],    //被后缀连接的数
nex[maxn * 2][maxc],  //状态转移(尾部加一个字符的下一个状态)(图)
idx, //节点编号
crp,
last;    //最后节点
ll epos[maxn * 2]; // enpos数(该状态子串出现数量)
char str[maxn];
ll a[maxn];		//长度为i的子串出现最大次数

void Iint() {	//初始化
	last = idx = crp = 1; //1表示root起始点 空集
	link[1] = len[1] = 0;
	memset(nex,0,sizeof(nex));
}
//SAM建图
void Extend(int c) {     //插入字符,为字符ascll码值
	int x = ++idx; //创建一个新节点x;
	len[x] = len[last] + 1; //  长度等于最后一个节点+1
	epos[x] = 1;  //接受节点子串除后缀连接还需加一
	int p;  //第一个有C转移的节点;
	for (p = last; p && !nex[p][c]; p = link[p])nex[p][c] = x;//沿着后缀连接 将所有没有字符c转移的节点直接指向新节点
	if (!p)link[x] = 1, cnt[1]++;  //全部都没有c的转移 直接将新节点后缀连接到起点
	else {
		int q = nex[p][c];    //p通过c转移到的节点
		if (len[p] + 1 == len[q])    //pq是连续的
			link[x] = q, cnt[q]++; //将新节点后缀连接指向q即可,q节点的被后缀连接数+1
		else {
			int nq = ++idx;   //不连续 需要复制一份q节点
			len[nq] = len[p] + 1;   //令nq与p连续
			link[nq] = link[q];   //因后面link[q]改变此处不加cnt
			memcpy(nex[nq], nex[q], sizeof(nex[q]));  //复制q的信息给nq
			for (; p&&nex[p][c] == q; p = link[p])
				nex[p][c] = nq;    //沿着后缀连接 将所有通过c转移为q的改为nq
			link[q] = link[x] = nq; //将x和q后缀连接改为nq
 			cnt[nq] += 2; //  nq增加两个后缀连接
		}
	}
	last = x;  //更新最后处理的节点
}
void GetNpos() {	//求npos数,即该节点字串出现次数
	queue<int>q;
	for (int i = 1; i <= idx; i++)
		if (!cnt[i])q.push(i);   //将所有没被后缀连接指向的节点入队
	while (!q.empty()) {
		int x = q.front();
		q.pop();
		epos[link[x]] += epos[x]; //子串数量等于所有后缀连接指向该节点的子串数量和+是否为接受节点
		if (--cnt[link[x]] == 0)q.push(link[x]);   //当所有后缀连接指向该节点的处理完毕后再入队
	}
}
void GetSubNum() {	//求不相同字串数量
	ll ans = 0;
	for (int i = 2; i <= idx; i++)ans += len[i] - len[link[i]];	//一状态子串数量等于len[i]-len[link[i]]
	printf("%lld\n",ans);
}
void GetSubMax() {	//求出所有长度为k的子串中出现次数最多的子串出现次数
	GetNpos();
	int n = strlen(str);
	for (int i = 1; i <= idx; i++)a[len[i]] = max(a[len[i]], epos[i]);	//长度≤k的子串中出现次数最多的子串出现次数的最小值
	for (int i = n - 1; i >= 1; i--)a[i] = max(a[i], a[i + 1]);		//求一遍后缀最大值就是答案
	for (int i = 1; i <= n; i++)printf("%lld\n", a[i]);
}
//匹配位置
int match(const char ch){
    return nex[crp][ch-'a'];
}
//删除
void withdraw(const int length){
    while(crp!=0&&len[link[crp]]>=length){
        crp=link[crp];
    }
    if(crp==0)crp=1;
}
void Transfer(const int length,const char ch){
    crp=nex[crp][ch-'a'];//S[crp].trans[ch-'a'];
    if( crp==0 ) crp=1;
    withdraw(length);
}
ll dp[maxn];
int P,Q;
void solve(){
    Extend(str[0]-'a');
    dp[0]=P;
    int l=1,r=0;
    for(int i=1;i<strlen(str);i++){
        ++r;
        dp[i]=dp[i-1]+P;
        while((!match(str[i])||r-l+1>(i+1)/2)&&l<=r){
            Extend(str[l++]-'a');
            withdraw(r-l);
        }
        Transfer(r-l+1,str[i]);
        if(l<=r){
            dp[i]=min(dp[i],dp[i-(r-l+1)]+Q);
        }
    }
    Iint();
    printf("%lld\n",dp[strlen(str)-1]);
}

signed main() {
    /*
	freopen("D:\\02.txt","r",stdin);
	freopen("D:\\out.txt","w",stdout);
	clock_t start,End;
	start=clock();
    */
	Iint();
    while(scanf("%s",str)!=EOF){
        scanf("%lld%lld",&P,&Q);
        solve();
    }
    //End=clock();
    //cout<<"Time used :"<<(double)(End-start)/CLOCKS_PER_SEC<<endl;
	return 0;
}

 

第二个模板对应于HDU6583的后缀自动机题,只需要一个匹配,奈何标程模板写得好,保存一哈

#include<bits/stdc++.h>

typedef long long ll;

using namespace std;

const int maxp=26+5;
const int maxn=400000+5;
//自动机节点
struct Suffix_Node{
    /*
    trans://状态转移(尾部加一个字符的下一个状态)(图)(对应nex)
    par://后缀链接(最短串前部减少一个字符所到达的状态)(对应link)
    len://最长子串的长度(该节点字串数量=len[x]-len[link[x]])
    */
	int trans[maxp],par,len;
	void Clear(){
		memset(trans,0,sizeof(trans));
		par=len=0;
	}
};

class Suffix_Automaton{
 private:
    /*
    size:节点编号
    */
	Suffix_Node S[maxn];
	int p,np,size,crp;
 public:
 Suffix_Automaton():p(1),np(1),size(1),crp(1){};
	const void Clear(){
		for(int i=0;i<=size;++i) S[i].Clear();
		p=np=size=crp=1;
	}
	//匹配
	const int Match(const char ch)const{
		return S[crp].trans[ch-'a'];
	}
	//删除
	const void Withdraw(const int len){
		while( crp!=0 && S[S[crp].par].len>=len ) crp=S[crp].par;
		if( crp==0 ) crp=1;
	}
	失配时,需要沿着失配边转移
	const void Transfer(const int len,const char ch){
		crp=S[crp].trans[ch-'a'];
		if( crp==0 ) crp=1;
		Withdraw(len);
	}
	const void Insert(const char ch){
		int x=ch-'a';
		np=++size;
		S[np].len=S[p].len+1;
		while( p!=0 && !S[p].trans[x] ) S[p].trans[x]=np,p=S[p].par;
		if( p==0 ) S[np].par=1;
		else{
			int q,nq;
			q=S[p].trans[x];
			if( S[q].len==S[p].len+1 ) S[np].par=q;
			else{
				nq=++size;
				S[nq]=S[q],S[nq].len=S[p].len+1;
				S[np].par=S[q].par=nq;
				while( p!=0 && S[p].trans[x]==q ) S[p].trans[x]=nq,p=S[p].par;
			}
		}
		p=np;
	}
}SAM;

string s;
ll f[maxn],p,q;

ll DP(){
	SAM.Insert(s[0]);
	f[0]=p;
	int l=1,r=0;
	for(int i=1;i<(int)s.size();++i){
		++r;
		f[i]=f[i-1]+p;
		while( ( !SAM.Match(s[i]) || r-l+1>(i+1)/2 ) && l<=r ){
			SAM.Insert(s[l++]);
			SAM.Withdraw(r-l);
		}
		SAM.Transfer(r-l+1,s[i]);
		if(l<=r){
			f[i]=min(f[i],f[i-(r-l+1)]+q);
		}
	}
	SAM.Clear();
	return f[s.size()-1];
}

int main(){
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	while( cin>>s>>p>>q){
		ll ans=DP();
		cout<<ans<<endl;
	}
	//cout<<"Time used :"<<(double)(End-start)/CLOCKS_PER_SEC<<endl;
	return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值