【Leetcode】459. Repeated Substring Pattern（配数学证明）

题目地址：

https://leetcode.com/problems/repeated-substring-pattern/

给定一个长$n$字符串$s$，问是否是某个字符串自我重复若干次得到的（这里若干次指的是多于$1$次）。

思路是用KMP算法。先求$s$的next数组$ne$，但这里数组需要多求一位，目的是求出$s$的最长相等前后缀的长度，设其为$l$。如果$l=0$，那么显然返回false；我们可以证明，当$l>0$时，返回true当且仅当$(n-l)|n$。

充分性：
若$(n-l)|n$，则$\frac{n}{n-l}\ge 2$，所以$n\le 2l$，设$x=n-l\le l$，所以$x|n$，由next数组定义知道，$s[0:x-1]=s[x:2x-1]=s[2x:3x-1]=...=s[n-x:n-1]$，也就是说$s$是$s[0:x-1]$自我重复若干次得到的。

必要性：
若$s=AA...A$，其中$A$代表某个字符串，也就是$s$可以表达为某个字符串自我重复$k$次。当$s$给定时，我们取最短的那个$A$，所以此时$k$也达到最大，有$n=k{l}_A$，则$ne[n]\ge (k-1)l_A$。下面我们证明$ne[n]\le (k-1)l_A$。反证法，如果$ne[n]\ge (k-1)l_A+1$，设$ne[n]=l$，$x=n-l$，由必要性证明知道，$s[0:x-1]=s[x:2x-1]=...=s[ux:(u+1)x-1]$，其中$n-1-((u+1)x-1)<x$（这个不等式一定是成立的，否则的话$s$就可以由$s[0:x-1]$重复若干次得到，这与$A$的最短性矛盾），所以有$$\begin{gathered} s[n-x:(u+1)x-1]=s[n-1-((u+1)x-1):x-1] \\ =s[n-(u+1)x:x-1] \end{gathered}$$这与$ne[n]=l$是矛盾的，因为$ne[n]$还可以拓展，应该有$ne[n]>l$才对。如此一来，$ne[n]$取任何大于$(k-1)l_A$的数都会造成矛盾，只能有$ne[n]=(k-1)l_A$。而$n-(k-1)l_A=l_A$，所以$(n-(k-1)l_A)|n$。证毕。

所以我们证明了$s$是由某字符串自我重复若干次得到的，当且仅当$l>0\wedge (n-l)|n$。并且顺便知道了，此时$s[0:l-1]$就是最短的可以重复若干次得到$s$的那个字符串。关于next数组怎么求，参考https://blog.csdn.net/qq_46105170/article/details/106168535。代码如下：

class Solution {
 public:
  bool repeatedSubstringPattern(string s) {
    int n = s.size();
    s = ' ' + s;
    int ne[n + 1];
    ne[1] = 0;
    for (int i = 2, j = 0; i <= n; i++) {
      while (j && s[i] != s[j + 1]) j = ne[j];
      if (s[i] == s[j + 1]) j++;
      ne[i] = j;
    }

    return ne[n] && n % (n - ne[n]) == 0;
  }
};

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