題目鏈接: K - Count the string HDU - 3336
給你一個字符串,枚舉其所有的前綴,輸出其所有前綴出現的次數
#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)(j);i <= (int)(k);i ++)
#define per(i,j,k) for(int i = (int)(j);i >= (int)(k);i --)
#define mmm(a,b) memset(a,b,sizeof(a))
#define push_back pb
typedef long long ll;
const int INF = (int)0x3f3f3f3f;
const int MAXN = (int)2e5+7;
const int MOD = (int)1e4+7;
char sub[MAXN];
int Next[MAXN];
int dp[MAXN];
int len1;
void MakeNext(){
Next[0] = 0;
for (int i = 1,j = 0;i < len1;i ++){
while (j > 0 && sub[i] != sub[j]) j = Next[j-1];
if (sub[j] == sub[i]) j ++;
Next[i] = j;
}
}
int main()
{
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
int T;
cin >> T;
while (T --){
cin >> len1 >> sub;
MakeNext();
dp[0] = 0;
int ans = 0;
rep(i,1,len1){
dp[i] = dp[Next[i-1]] + 1;
ans = (ans + dp[i])%MOD;
}
cout << ans << endl;
}
}
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。