[CF1538E] Funny Substrings （模拟）

题面

该场 Div. 3 最“难”的一道题：Funny Substrings
$\mathtt{O}\mathtt{I}\mathtt{D}$ 队长喜欢玩字符串，因为 $“\mathtt{O}\mathtt{n}\mathtt{e}\mathtt{I}\mathtt{n}\mathtt{D}\mathtt{a}\mathtt{r}\mathtt{k}”$ 是一个望而生畏的字符串。

$\mathtt{O}\mathtt{I}\mathtt{D}$ 会进行 $\mathtt{N}$ 次操作，每次操作形如以下两种之一：

• x := S : x 是变量名称，S 是一个具体的字符串，:= 符号两边都有空格。$\mathtt{O}\mathtt{I}\mathtt{D}$ 把此时的 x 叫做 $SY$ 变量。该操作意思是把 x 赋值为字符串 S。
• x = a + b : x,a,b 是变量名称，= 符号和 + 符号两边都有空格。$\mathtt{O}\mathtt{I}\mathtt{D}$ 把此时的 x 也叫做 $SY$ 变量。该操作意思是把 x 赋值为 a 和 b 首尾拼接起来的字符串。保证 a 和 b 在此操作之前作为 $SY$ 变量出现过。

现在问你，在最后一次出现的 $SY$ 变量储存的字符串中，字符串 haha 作为子串出现过多少次。比如：haha 在 hahahaha 中出现了 3 次。

一共 $\mathtt{T}\le 10^3$ 组数据，每组数据中 $1\le \mathtt{N}\le 50$，$1\le |\mathtt{S}|\le 5$ ，$1\le |变量名|\le 5$。

题解

(我看这道题过的人最少，姑且认为它难吧)

不难发现，最终的字符串长度可能超过 long long 范围，因此，直接 string 模拟不是个好方法。

我们会发现，对于一个字符串，有用的信息可以整理成一个三元组 $\{\mathtt{l}(\mathtt{s}\mathtt{t}\mathtt{r}\mathtt{i}\mathtt{n}\mathtt{g}),\mathtt{x}(\mathtt{l}\mathtt{o}\mathtt{n}\mathtt{g}\mathtt{l}\mathtt{o}\mathtt{n}\mathtt{g}),\mathtt{r}(\mathtt{s}\mathtt{t}\mathtt{r}\mathtt{i}\mathtt{n}\mathtt{g})\}$，分别表示：该字符串最左边 3 个(也许不足 3 个)字符，该字符串中 haha 的出现次数，该字符串最右边 3 个(也许不足 3 个)字符。

这个不难处理。关键是这样的三元组可以方便地定义 a + b 操作，返回另一个三元组：

• 把 ${\mathtt{x}}_{\mathtt{a}}$ 和 ${\mathtt{x}}_{\mathtt{b}}$ 加起作为 $\mathtt{x}$ ，再把 ${\mathtt{r}}_{\mathtt{a}}$ 和 ${\mathtt{l}}_{\mathtt{b}}$ 拼起来(长度不超过 6)，统计里面的 haha 数量，也加到 $\mathtt{x}$ 里面去。
• 把 $\mathtt{l}$ 赋为 ${\mathtt{l}}_{\mathtt{a}}$ ，如果长度不足 3，再把 ${\mathtt{l}}_{\mathtt{b}}$ 拼在 $\mathtt{l}$ 后面，把长度超过 3 的砍掉。
• 把 $\mathtt{r}$ 赋为 ${\mathtt{r}}_{\mathtt{b}}$ ，如果长度不足 3，再把 ${\mathtt{r}}_{\mathtt{a}}$ 拼在 $\mathtt{r}$ 前面，把长度超过 3 的砍掉。

这样就做出这道模拟题了，中间的字符串操作，可以暴力用 string 或者 char*+sprintf，时间足够。对于每个变量怎么存三元组，你可以用哈希也可以用 map。

CODE

#include<set>
#include<map>
#include<queue>
#include<ctime>
#include<cmath>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define MAXN 105
#define ENDL putchar('\n')
#define LL long long
#define DB double
#define lowbit(x) ((-x) & (x))
#define eps 1e-9
LL read() {
	LL f = 1,x = 0;char s = getchar();
	while(s < '0' || s > '9') {if(s=='-')f = -f;s = getchar();}
	while(s >= '0' && s <= '9') {x=x*10+(s-'0');s = getchar();}
	return f * x;
}
int n,m,i,j,s,o,k;
struct it{
	char l[5],r[5];
	LL nm;it(){memset(l,0,sizeof(l));memset(r,0,sizeof(r));nm=0;}
	it(char *L,char *R,LL N){
		strncpy(l,L,3);nm=N;memset(r,0,sizeof(r));
		int le = strlen(R);
		for(int i = max(le-3,0),j=0;i < le;i ++,j ++) r[j]=R[i];
	}
};
it operator + (it a,it b) {
	char ss[10]; int ls; LL as = a.nm+b.nm;
	sprintf(ss,"%s%s",a.r,b.l);
	ls = strlen(ss);
	for(int i = 0;i <= ls-4;i ++) if(ss[i]=='h'&&ss[i+1]=='a'&&ss[i+2]=='h'&&ss[i+3]=='a') as ++;
	char rel[10],rer[10];
	if(strlen(a.l) < 3) sprintf(rel,"%s%s",a.l,b.l);
	else strcpy(rel,a.l);
	if(strlen(b.r) < 3) sprintf(rer,"%s%s",a.r,b.r);
	else strcpy(rer,b.r);
	return it(rel,rer,as);
}
map<string,it> mp;
int main() {
	int T = read();
	while(T --) {
		n = read();
		mp.clear();
		string las;
		for(int i = 1;i <= n;i ++) {
			string A,md,B,C;
			cin>>A;
			las = A;
			cin>>md;
			if(md[0]==':') {
				char ss[15]; int le,as = 0;
				scanf("%s",ss); le = strlen(ss);
				for(int i = 0;i <= le-4;i ++) {
					 if(ss[i]=='h'&&ss[i+1]=='a'&&ss[i+2]=='h'&&ss[i+3]=='a') as ++;
				}
				mp[A] = it(ss,ss,(LL)as);
			}
			else {
				cin>>B;cin>>md;cin>>C;
				mp[A] = mp[B] + mp[C];
			}
		}
		printf("%lld\n",mp[las].nm);
	}
	return 0;
}