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