牛客小白月赛18 B.Forsaken喜欢字符串 题解（哈希）

题目链接

题目思路

前言
看到题目好多个累加号，一下就蒙了，其实仔细看一下还是可以明白题意得

正题
对于每个询问，本质上是求其他所有串的长度为len的子串是串x的子串的个数，然后总数乘上len就是答案,注意到k特别小，因此我们只需要存下每个串的所有哈希值，并且记录这个哈希值出现的次数，然后每次查询哈希值出 现的次数就行了。复杂度是O(nk^2 +qk)

代码

直接stl

#include<iostream>
#include<map>
using namespace std;
typedef long long ll;
const int maxn=5e4+5;
int n,k,q,x,len;
ll ans;
string s[maxn];
map<string,int> cnt;
int main(){
	ios::sync_with_stdio(false);//加速读入 
	cin.tie(0),cout.tie(0);
	cin>>n>>k;
	for(int i=1;i<=n;i++){
		cin>>s[i];
		for(int len=1;len<=k;len++){//字串的长度 
			for(int j=0;j+len<=k;j++){//起点 
				cnt[s[i].substr(j,len)]++;
			}
		}
	}
	cin>>q;
	while(q--){
		ans=0;
		cin>>x>>len;
		map<string,int> me;//重新定义了则不需要清空 
		for(int i=0;i+len<=k;i++){
			me[s[x].substr(i,len)]++;
		}
		for(int i=0;i+len<=k;i++){
			ans+=cnt[s[x].substr(i,len)]-me[s[x].substr(i,len)];	
		}
		cout<<ans*len<<endl;
	}
	return 0;
}