ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

CSUSTOJ 2002 Banana(回文树)

2021-07-30 22:00:41  阅读:196  来源: 互联网

标签:cnt End CSUSTOJ int ++ 2002 Start -- Banana


题目链接
在这里插入图片描述
利用回文树求前缀后缀的回文子串数,枚举计算答案即可。


int main() {
    #ifdef LOCAL
        INPUT;
        OUTPUT;
    #endif

   // scanf("%d", &n);
    scanf("%s", S + 1);
n=strlen(S+1);
    LL cnt = 0;
    memset(R, 0, sizeof R);
    for (int i = 1, l = 0, r = 0; i <= n; ++i) {
        if (i <= r) R[i] = min(r - i + 1, R[l + r - i]);
        while (1 <= i - R[i] && i + R[i] <= n && S[i - R[i]] == S[i + R[i]]) R[i]++;
        if (i + R[i] - 1 > r) l = i - R[i] + 1, r = i + R[i] - 1;
        cnt += R[i];
        Start[i - R[i] + 1]++; End[i]++;
        Start[i + 1]--; End[i + R[i]]--;
        //printf("%d ", R[i]);
    }
    //puts("");

    memset(R, 0, sizeof R);
    for (int i = 1, l = 0, r = 0; i <= n - 1; ++i) {
        if (i < r) R[i] = min(r - i, R[l + r - i - 1]);
        while (1 <= i - R[i] && i + R[i] < n && S[i - R[i]] == S[i + R[i] + 1]) R[i]++;
        if (i + R[i] > r) l = i - R[i] + 1, r = i + R[i];
        cnt += R[i];
        Start[i - R[i] + 1]++; End[i + 1]++;
        Start[i + 1]--; End[i + R[i] + 1]--;
        //printf("%d ", R[i]);
    }
    //puts("");

    LL ans =0;
    for (int i = 1; i <= n; ++i) Start[i] = (Start[i] + Start[i - 1]) % M;
    for (int i = 1; i <= n; ++i) End[i] = (End[i] + End[i - 1]) % M;
    for (int i = 1; i <= n; ++i) End[i] = (End[i] + End[i - 1]) % M;

    for (int i = 1; i <= n; ++i) ans = (ans + Start[i] * End[i - 1] % M + M) % M;
    cout << ans;
    return 0;
}

标签:cnt,End,CSUSTOJ,int,++,2002,Start,--,Banana
来源: https://blog.csdn.net/weixin_45948940/article/details/119256698

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有