当前位置:
首页 >
2021牛客OI赛前集训营-提高组(第五场)D-牛牛的border【SAM】
发布时间:2023/12/3
67
豆豆
生活随笔
收集整理的这篇文章主要介绍了
2021牛客OI赛前集训营-提高组(第五场)D-牛牛的border【SAM】
小编觉得挺不错的,现在分享给大家,帮大家做个参考.
正题
题目链接:https://ac.nowcoder.com/acm/contest/20110/D
题目大意
求一个长度为nnn的字符串的所有子串的borderborderborder长度和。
1≤n≤1051\leq n\leq 10^51≤n≤105
解题思路
考虑到两个相同的子串会作为一个子串的borderborderborder,所以问题可以变为求所有相同子串对的长度之和。
然后直接跑出SAMSAMSAM然后对于每个节点统计它在字符串里的出现次数,然后所有的lenfa+1∼lenxlen_{fa+1}\sim len_xlenfa+1∼lenx都是这个节点的字符串长度,用过等比序列求和就好了。
时间复杂度:O(n)O(n)O(n)(不算快速排序)
code
#include<cstdio> #include<cstring> #include<algorithm> #define ll long long using namespace std; const ll N=2e5+10; ll n,cnt,last,ans,p[N],len[N],fa[N],c[N],ch[N][26]; char s[N]; void Insert(char c){ll p=last,np=last=++cnt;len[np]=len[p]+1;for(;!ch[p][c];p=fa[p])ch[p][c]=np;if(!p)fa[np]=1;else{ll q=ch[p][c];if(len[p]+1==len[q])fa[np]=q;else{ll nq=++cnt;len[nq]=len[p]+1;memcpy(ch[nq],ch[q],sizeof(ch[nq]));fa[nq]=fa[q];fa[q]=fa[np]=nq;for(;ch[p][c]==q;p=fa[p])ch[p][c]=nq;}}return; } bool cmp(ll x,ll y) {return len[x]>len[y];} ll calc(ll l,ll r){if(!r)return 0;return (r+l)*(r-l+1)/2; } signed main() {scanf("%lld",&n);scanf("%s",s+1);cnt=last=1;for(ll i=1;i<=n;i++)Insert(s[i]-'a'),c[last]++;for(ll i=1;i<=cnt;i++)p[i]=i;sort(p+1,p+1+cnt,cmp);for(ll i=1;i<=cnt;i++)c[fa[p[i]]]+=c[p[i]];for(ll i=1;i<=cnt;i++){ll w=calc(len[fa[i]]+1,len[i]);ans+=calc(1,c[i]-1)*w;}printf("%lld\n",ans);return 0; }总结
以上是生活随笔为你收集整理的2021牛客OI赛前集训营-提高组(第五场)D-牛牛的border【SAM】的全部内容,希望文章能够帮你解决所遇到的问题。
- 上一篇: 不成功便成仁这句话什么意思 不成功便成仁
- 下一篇: P1852-跳跳棋【思维,差分,二分】