2020 Multi-University Training Contest 8 I
有一个长度为N的只包含小写字符的字符串,现在问你,能否将这个字符串分成K个等长的字符串,使得每个字符串作为字符串环的时候,他们是相等的。
这题对unordered_map好像不是很友好,我开vector来存就过了,不然就TLE。
然后,我们可以对所有的因子处理出来,然后,对于枚举长度为len的时候,只需要把第一个串的len种字符串环的哈希值都存进去就可以了,后面的几个字符串只需要直接判断就可以了。
当然,这道题还可以用最小表示法来做,如果被卡哈希了,可以试一下。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <bitset>
#include <unordered_map>
#include <unordered_set>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
using namespace std;
typedef unsigned long long ull;
typedef unsigned int uit;
typedef long long ll;
const int maxN = 5e6 + 7;
const ll hs = 233, mod = 1e7 + 7;
ll Bas[maxN], h[maxN];
int N;
char s[maxN];
vector<int> vt, cop;
vector<bool> mp;
inline ll _Get(int l, int r) { return (h[r] - h[l - 1] * Bas[r - l + 1] % mod + mod) % mod; }
int main()
{
mp.resize(mod + 1);
Bas[0] = 1;
for(int i=1; i<=5000000; i++) Bas[i] = Bas[i - 1] * hs % mod;
int T; scanf("%d", &T);
while(T--)
{
scanf("%d", &N);
scanf("%s", s + 1);
if(N == 1) { printf("No\n"); continue; }
h[0] = 0;
for(int i=1; i<=N; i++) h[i] = (h[i - 1] * hs % mod + s[i]) % mod;
vt.clear();
vt.push_back(1);
for(int i=2; i * i <= N; i++)
{
if(N % i == 0)
{
vt.push_back(i);
if(i * i != N) vt.push_back(N / i);
}
}
int len = (int)vt.size();
ll tmp;
bool ok = true;
for(int i=0, k; i<len; i++)
{
k = vt[i];
mp.clear();
mp[h[k]] = true;
cop.push_back((int)h[k]);
for(int j=2; j<=k; j++)
{
tmp = (_Get(j, k) * Bas[j - 1] % mod + _Get(1, j - 1)) % mod;
mp[tmp] = true;
cop.push_back((int)tmp);
}
ok = true;
for(int j=k + 1; ok && j<=N; j+=k)
{
tmp = _Get(j, j + k - 1);
if(!mp[tmp]) ok = false;
}
while(!cop.empty())
{
mp[cop.back()] = false;
cop.pop_back();
}
if(ok) break;
}
printf(ok ? "Yes\n" : "No\n");
}
return 0;
}
/*
2
8
bbcaabbc
10
ababbaaabb
ans:
Yes
No
*/