前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >HDU 4416 后缀自动机

HDU 4416 后缀自动机

作者头像
用户2965768
发布2019-08-01 11:05:26
2280
发布2019-08-01 11:05:26
举报
文章被收录于专栏:wymwym

题意:多组样例t,每个样例一个数n,接下来一个字符串 T ,n个字符串S,问T的子串有多少没有在S中出现

解:先将n个字符串加入后缀自动机,统计子串个数 ans,再把T加入后缀自动机,统计字符串个数ans2,

ans2 - ans1 就是答案

代码语言:javascript
复制
#include <bits/stdc++.h>
#define ll long long
const int maxn = 1000008;
const int maxc = 28;
using namespace std;
struct Suffix_Automaton {
	int len[maxn * 2], //最长子串的长度(该节点子串数量=len[x]-len[link[x]])
	    link[maxn * 2],   //后缀链接(最短串前部减少一个字符所到达的状态)
	    nex[maxn * 2][maxc],  //状态转移(尾部加一个字符的下一个状态)(图)
	    idx, //结点编号
	    last;    //最后结点


	void init() {	//初始化
	for(int i=1;i<=idx;i++)
		link[i] = len[i] = 0,memset(nex[i],0,sizeof(nex[i]));
		last = idx = 1; //1表示root起始点 空集
	}
//SAM建图
	void extend(int c) {     //插入字符,为字符ascll码值
		int x = ++idx; //创建一个新结点x;
		len[x] = len[last] + 1; //  长度等于最后一个结点+1
		//num[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;
		else {
			int q = nex[p][c];    //p通过c转移到的结点
			if (len[p] + 1 == len[q])    //pq是连续的
				link[x] = q;
			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
			
			}
		}
		last = x;  //更新最后处理的结点
	} 

	ll getSubNum() {	//求不相同子串数量
		ll ans = 0;
		for (int i = 2; i <= idx; i++)ans += len[i] - len[link[i]];	//一状态子串数量等于len[i]-len[link[i]]
		return ans;
	} 

} sam;
string str1,str2;
int main()
{
	cin.sync_with_stdio(0);
	int T,n,lenn; 
	int ccnt = 1;
	cin>>T;
	while(T--)
	{
		sam.init();
		cin>>n;
		cin>>str1;
		for(int i=1;i<=n;i++){
			sam.last = 1;
			cin>>str2; lenn = str2.size();
			for(int j=0;j<lenn;j++){
				sam.extend(str2[j]-'a');
			}
		}
		ll ans = sam.getSubNum();	
		sam.last = 1;
		lenn = str1.size();
		for(int i=0;i<lenn;i++){
			sam.extend(str1[i]-'a');
		}
		ans = sam.getSubNum() - ans;
		printf("Case %d: %lld\n",ccnt++,ans);
	}
	return 0;
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019年07月27日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档