E. Funny Substrings （div3）字符串模拟

最新推荐文章于 2024-10-20 01:52:41 发布

原创最新推荐文章于 2024-10-20 01:52:41 发布 · 114 阅读

0 ·

CC 4.0 BY-SA版权

字符串专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种名为E.FunnySubstrings的算法，针对字符串处理中的模式串计算进行了优化，通过合并策略减少重复计算，主要关注字符串后缀与前缀的结合。作者展示了如何利用C++实现并利用map和字符串操作加速模式数量的查找。

E. Funny Substrings
注意到我们的模式串是"haha",长度为4，我们算出的结果没必要保存全部的字符，我们只需保存前三个字符和后三个字符就行了，当合并的时候，我们让前一个字符串的后缀与后一个字符串的前缀合并，求一次模式串的数量，再加上两个字符串之前就已经算出来的自己的模式串的数量，就是合并之后的数量。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 9 ;
ll t, n;
string s;
map <string,string> ma;
map <string, ll> ans;
string getfirst(string t)
{
	if(t.size() < 3) return t;
	else return t.substr(0, 3);
}
string getlast(string t)
{
	if(t.size() < 3) return t;
	else return t.substr(t.size() - 3, 3);
}
ll count(string s,  string t)
{
	ll cnt = 0;
	for(int i = 0; i + t.size() <= s.size(); ++i)
		if(s.substr(i, t.size()) == t) ++cnt;
	return cnt;
}
int main()
{
	cin >> t;
	while(t--)
	{
		ma.clear(); ans.clear();
		scanf("%d", &n);
		ll tmp = 0;
		for(int i = 1; i <= n; ++i)
		{
			string a, b, c, d, e;
			cin >> a >> b >> c;
			if(b == ":=")
			{
				ans[a] = count(c, "haha");
				ma[a] = c;
			}
			else 
			{
				cin >> d >> e;
				string f = ma[c] + ma[e];
				ans[a]=ans[c]+ans[e]+count(getlast(ma[c])+getfirst(ma[e]),"haha");
				if(f.size() >= 7) // 缩小一下字符
					f = getfirst(f)+"@"+getlast(f);
				ma[a] = f;
			}
			tmp = ans[a];
		}
		cout << tmp << endl;
	}
    return 0;
}