E. Funny Substrings (div3) 字符串模拟

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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值