2020牛客多校二 A. All with Pairs (hash+next数组)

题意：
题解：hash+next数组
先把所有后缀哈希，用map存一下。
接下来遍历所有字符串的所有前缀，累加对应前缀哈希map。
因为总长度是1e6，不会t。

但是有一个问题，就是累加的时候会重复，如字符串自身aba，a算了一次，aba算了一次，但我们要取最长的，所以要把a去掉。

这时候就要用到next数组，我们用$cnt[i]$表示长度为$i$的前缀有$cnt[i]$个后缀与之相等，再遍历一次进行$cnt[next[i]]-=cnt[i]$操作即可。

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
#define ull unsigned ll
using namespace std;
const int maxn = 1e6 + 5;
const int mod = 998244353;
const ull ba = 2333;
int n;
string s[maxn];
ll cnt[maxn];
int nxt[maxn];
map<ull, int> ma;
void getHash(string x) {
	int len = x.length();
	ull ans = 0, po = 1;
	for (int i = len - 1; i >= 0; i--) {
		ans = ans + x[i] * po;
		po = po * ba;
		ma[ans]++;
	}
}
void getNext(string s, int Next[]) {  //下标从0开始
	int len = s.size();
	int k = -1;
	Next[0] = k;
	for (int i = 1; i < len; i++) {
		while (k > -1 && s[k + 1] != s[i]) k = Next[k];
		if (s[k + 1] == s[i]) k++;
		Next[i] = k;
	}
}
int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> s[i];
		getHash(s[i]);
	}
	ll ans = 0;
	for (int i = 1; i <= n; i++) {
		ull temp = 0;
		getNext(s[i], nxt);
		for (int j = 0; j < s[i].length(); j++) {
			temp = temp * ba + s[i][j];
			cnt[j] = ma[temp];
		}
		for (int j = 0; j < s[i].length(); j++) {
			if (nxt[j] == -1) continue;
			cnt[nxt[j]] -= cnt[j];
		}
		for (int j = 0; j < s[i].length(); j++) {
			ans = (ans + 1ll * cnt[j] * (1ll * j + 1) % mod * (1ll * j + 1) % mod) % mod;
		}
	}
	printf("%lld\n", ans);
	return 0;
}