标签：while color cst 题意 hdoj3336 get return next oid
题目链接：https://vjudge.net/problem/HDU-3336
题意：给定长为n(<=2e5)的字符串s，求s的每个前缀在s中出现的次数之和。
思路：
用dp[i]表示以s[i]为结尾的子串是s的某一种前缀的方案数，那么dp[i]=dp[nex[i]]+1，因为[nex[i]-(i-nex[i])+1 , nex[i]] 和 [nex[i]+1 , i] 是一样的，所以计算后者时只用把前者+1即可（+1是因为以[1,i]为前缀的个数加一）。
AC代码：
#include<cstdio> #include<algorithm> #include<cstring> using namespace std; const int maxn=2e5+5; const int MOD=1e4+7; int T,n,ans,nex[maxn],dp[maxn]; char s[maxn]; void get_next(){ int j; j=nex[0]=-1; for(int i=1;i<n;++i){ while(j>-1&&s[i]!=s[j+1]) j=nex[j]; if(s[i]==s[j+1]) ++j; nex[i]=j; } } int main(){ scanf("%d",&T); while(T--){ scanf("%d%s",&n,s); get_next(); ans=0; for(int i=1;i<=n;++i){ dp[i]=(dp[nex[i-1]+1]+1)%MOD; ans=(ans+dp[i])%MOD; } printf("%d\n",ans); } return 0; }
标签：while color cst 题意 hdoj3336 get return next oid
原文地址：https://www.cnblogs.com/FrankChen831X/p/11787659.html
