【ACWing】777. 字符串乘方

最新推荐文章于 2025-06-13 15:32:45 发布

记录算法题解

最新推荐文章于 2025-06-13 15:32:45 发布

阅读量4.2k

点赞数 8

CC 4.0 BY-SA版权

分类专栏： AC 数组、字符串、链表与模拟文章标签：算法 c++

本文链接：https://blog.csdn.net/qq_46105170/article/details/120029075

AC 数组、字符串、链表与模拟专栏收录该内容

252 篇文章

订阅专栏

该博客介绍了一种利用KMP算法求解字符串乘方的问题。给定一个字符串s，需要找到最大的整数n，使得存在一个字符串a，满足s等于a的n次方。代码实现中，首先构建了KMP的next数组，然后通过判断s是否能被自身的一个子串整除来确定n的值。输出的时间复杂度为O(l_s)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目地址：

https://www.acwing.com/problem/content/description/779/

给定两个字符串 $a$ 和 $b$ ，我们定义 $a \times b$ 为他们的连接。例如，如果 $a = a b c$ 而 $b = d e f$ ，则 $a \times b = a b c d e f$ 。如果我们将连接考虑成乘法，一个非负整数的乘方将用一种通常的方式定义： $a^0=""$ (空字符串)， $a^{n+1}=a×(a^n)$ 。

输入格式：
输入包含多组测试样例，每组测试样例占一行。每组样例包含一个字符串 $s$ ， $s$ 的长度不超过 $100$ 。最后的测试样例后面将是一个点号作为一行。

输出格式：
对于每一个 $s$ ，你需要输出最大的 $n$ ，使得存在一个字符串 $a$ ，让 $s=a^n$ 。

可以用KMP算法。参考https://blog.csdn.net/qq_46105170/article/details/106184933。代码如下：

#include <iostream>
using namespace std;

const int N = 110;
string s;
int ne[N];

void build_next() {
    for (int i = 0, j = ne[0] = -1; i <= s.size(); ) {
        if (j == -1 || s[i] == s[j]) ne[++i] = ++j;
        else j = ne[j];
    }
}

int main() {
    while (cin >> s, s != ".") {
        build_next();
        int n = s.size(), len = ne[n];
        if (n % (n - len) == 0) printf("%d\n", n / (n - len));
        else puts("1");
    }

    return 0;
}

时空复杂度 $O(l_s)$ 。