P3804 【模板】后缀自动机(SAM)

题目描述

给定一个只包含小写字母的字符串 S。

请你求出 S 的所有出现次数不为 11 的子串的出现次数乘上该子串长度的最大值。

输入格式

一行一个仅包含小写字母的字符串 S。

输出格式

一个整数,为所求答案。

题解:这里就不讲后缀自动机的模板相关问题了,网有很多关于后缀自动机的详解(推荐)。我在看完了后缀自动机之后做这题,卡住我的不是后缀自动机,而是以下操作

 

	//Part 2 按len从大到小排序(和SA好像啊)后计算答案
	for(int i=1;i<=node;i++) t[len[i]]++;
	for(int i=1;i<=node;i++) t[i]+=t[i-1];
	for(int i=1;i<=node;i++) A[t[len[i]]--]=i;
	for(int i=node;i>=1;i--)
	{//从小到大枚举,实际上在模拟parent树的DFS
		int now=A[i];siz[fa[now]]+=siz[now];
		if(siz[now]>1) ans=max(ans,1ll*siz[now]*len[now]);
	}

 这段代码的要做的就是把节点 按节点代表最大字符串长度 排序,但我一直看不懂这里的代码逻辑,于是转换方式,用容易看懂的逻辑写代码。

方式一

struct edge
{
	int pos, len;
	bool operator<(edge e1)
	{
		return len > e1.len;
	}
}e[N];//存节点及其代表的最大长度


for (int i = 1; i <= tot; i++)
{
	e[i] = { i,nd[i].len };
}

sort(e + 1, e + tot + 1);//按长度从大到小排序
//因为节点所代表的字符长度越长,这个节点就越接近于叶子节点
//所以排好序后,从小到大遍历也就是从叶子节点到根节点的遍历,把endpos的个数往上累加

for (int i = 1; i <= tot; i++)
{

	siz[nd[e[i].pos].fa] += siz[e[i].pos];
}

for (int i = 1; i <= tot; i++)
{
	if (siz[i] > 1)
		ans = max(ans, (long long)(siz[i] * nd[i].len));
}

 方式二

int deg[N << 2];
queue<int> q;
void solve()
{
	while (!q.empty()) q.pop();
	memset(deg, 0, sizeof deg);
	for (int i = 1; i <= num; i++)
	{
		++deg[a[i].fa];
		siz[i] = 0;
	}

	for (int i = 1; i <= num; i++)
	{
		
		if (!deg[i]) q.push(i);//不是任何节点的父节点,即叶子节点入队
	}
	while (!q.empty())
	{
		int x = q.front();
		q.pop();
		if (pre[x])
			++siz[x];
		if (a[x].fa == 1) continue;
		siz[a[x].fa] += siz[x];
		if ((--deg[a[x].fa]) == 0)//该节点的子节点计算过一次,减一,当减为0后,即子节点全部计            
                                  //算完成,视为叶子节点入队
		{
			q.push(a[x].fa);
		}
	}
	long long ans = 0;
	for (int i = 1; i <= num; i++)
	{
		if (siz[i] > 1)
		{
			ans = max(ans, (long long)(siz[i] * a[i].len));
		}
	}

	cout << ans;
}

现在贴上这三种方式的完整代码

一,

#include<iostream>
#include<map>
#include<algorithm>	
#include<cmath> 
#include<cstring>
#include<stack>
#include<vector>
#include<queue>
using namespace std;
const long long N = 2e6 + 115;
struct node
{
	int fa, len;
	int son[26];
	long long size;
	node()
	{
		fa = len = 0;
		size = 0;
		memset(son, 0, sizeof son);
	}
}d[N];

int las, tot,n;

void add(int now)
{
	int p = las;
	int np  = ++tot;
	las = np;
	d[np].size = 1;//作为实节点,原始endpos=1
	d[np].len = d[p].len + 1;
	for (; p && !d[p].son[now]; p  = d[p].fa) d[p].son[now] = np;
	if (!p) d[np].fa = 1;
	else
	{
		int q = d[p].son[now];
		if (d[q].len == d[p].len + 1) d[np].fa = q;
		else
		{
			int nq = ++tot;
			d[nq] = d[q];
			d[nq].size = 0;//作为虚节点,原始endpos数=0
			d[nq].len = d[p].len + 1;
			d[q].fa = nq;
			d[np].fa = nq;
			for (; p && d[p].son[now] == q; p = d[p].fa) d[p].son[now] = nq;
		}
	}
}

int tmp[N], tp[N];

void so()
{
	for (int i = 0; i <= n; i++)
		tmp[i] = 0;
	for (int i = 1; i <= tot; i++)
		tmp[d[i].len]++;
	for (int i = 1; i <=n ; i++)
		tmp[i] += tmp[i - 1];
	for (int i = 1; i <=tot; i++)
		tp[tmp[d[i].len]--] = i;
	for (int i = tot; i >= 1; i--)
	{
		int now = tp[i];
		d[d[now].fa].size += d[now].size;
	}
}


void getans()
{
	long long ans = 0;
	for (int i = 1; i <= tot;i++)
	{
		
		if (d[i].size > 1)
			ans = max(ans, d[i].size * d[i].len);
	}
	cout << ans;
}

int main()
{
	string s;
	cin >> s;
	n = s.size();
	las = tot = 1;
	for (int i = 0; i < n; i++)
	{
		add(s[i] - 'a');
	}
	so();
	getans();
}

 

 二,

#include<iostream>
#include<map>
#include<algorithm>	
#include<cmath> 
#include<cstring>
#include<stack>
#include<vector>
using namespace std;
const long long N = 2e6 + 115;

int siz[N];

struct node
{
	int fa, len;
	int ch[26];
	node() { memset(ch, 0, sizeof(ch)); len = 0; }
}nd[N ];
int las = 1, tot = 1;
//las最近的一个实节点,tot节点总数 
struct edge
{
	int pos, len;
	bool operator<(edge e1)
	{
		return len > e1.len;
	}
}e[N];
void add(int c)
{
	int p = las, np;
	las = np = ++tot;
	siz[tot] = 1;
	nd[np].len = nd[p].len + 1;
	for (; p && !nd[p].ch[c]; p = nd[p].fa) nd[p].ch[c] = np;
	if (!p) nd[np].fa = 1;
	else
	{
		int q = nd[p].ch[c];
		if (nd[p].len + 1 == nd[q].len)
		{
			
			nd[np].fa = q;
		}
		else
		{
			int nq = ++tot;
			nd[nq] = nd[q];
			nd[nq].len = nd[p].len + 1;
			nd[q].fa = nd[np].fa = nq;
			for (; p && nd[p].ch[c] == q; p = nd[p].fa) nd[p].ch[c] = nq;
		}
	}
}

long long ans = 0;

int main()
{
	string s;
	cin >> s;
	int le = s.size();

	for (int i = 0; i < le; i++)
	{
		add(s[i] - 'a');
	}
	for (int i = 1; i <= tot; i++)
	{
		e[i] = { i,nd[i].len };
	}
	sort(e + 1, e + tot + 1);
	for (int i = 1; i <= tot; i++)
	{

			siz[nd[e[i].pos].fa] += siz[e[i].pos];
	}

	for (int i = 1; i <= tot; i++)
	{
		if (siz[i] > 1)
			ans = max(ans, (long long)(siz[i] * nd[i].len));
	}

	cout << ans;

}

三,

#include<iostream>
#include<map>
#include<algorithm>	
#include<cmath> 
#include<cstring>
#include<stack>
#include<vector>
#include<queue>
using namespace std;
const long long N = 1e6 + 115;

int siz[N << 2];

struct node
{
	int fa, len;
	int ch[26];
	node() { memset(ch, 0, sizeof(ch)); len = 0; }
}a[N << 2];

int cnt[N << 2];

int num = 1, np = 1;
bool pre[N << 2];
void add(int c)
{
	int p = np; np = ++num;
	pre[np] = true;//标记实节点
	a[np].len = a[p].len + 1;
	for (; p && !a[p].ch[c]; p = a[p].fa) a[p].ch[c] = np;
	if (!p) a[np].fa = 1;
	else
	{
		int q = a[p].ch[c];
		if (a[q].len == a[p].len + 1) a[np].fa = q;
		else
		{
			int nq = ++num;
			a[nq] = a[q];
			a[nq].len = a[p].len + 1;
			a[q].fa = a[np].fa = nq;
			for (; p && a[p].ch[c] == q; p = a[p].fa) a[p].ch[c] = nq;
		}
	}
}

int deg[N << 2];
queue<int> q;
void solve()
{
	while (!q.empty()) q.pop();
	memset(deg, 0, sizeof deg);
	for (int i = 1; i <= num; i++)
	{
		++deg[a[i].fa];
		siz[i] = 0;
	}

	for (int i = 1; i <= num; i++)
	{
		
		if (!deg[i]) q.push(i);//不是任何节点的父节点,即叶子节点入队
	}
	while (!q.empty())
	{
		int x = q.front();
		q.pop();
		if (pre[x])
			++siz[x];
		if (a[x].fa == 1) continue;
		siz[a[x].fa] += siz[x];
		if ((--deg[a[x].fa]) == 0)//该节点的子节点计算过一次,减一,当减为0后,即子节点全部计            
                                  //算完成,视为叶子节点入队
		{
			q.push(a[x].fa);
		}
	}
	long long ans = 0;
	for (int i = 1; i <= num; i++)
	{
		if (siz[i] > 1)
		{
			ans = max(ans, (long long)(siz[i] * a[i].len));
		}
	}

	cout << ans;
}

int main()
{
	string s;
	cin >> s;
	int le = s.size();

	for (int i = 0; i < le; i++)
	{
		add(s[i] - 'a');
	}

	solve();

}

 

特别注意:节点里的数组是整数数组,而不是字符数组,int ch[26] 而不是char ch[26] ,你猜我为什么着重提这个.....

 

 

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

linalw

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值