【Leetcode】1071. Greatest Common Divisor of Strings（配数学证明）

题目地址：

https://leetcode.com/problems/greatest-common-divisor-of-strings/

给定两个字符串$s$和$t$，如果存在字符串$x$使得$s=x+x+...+x$并且$t=x+x+...+x$，也就是说$s$和$t$都可以分割为某个相同字符串的拼接，那就返回最长满足条件的$x$，否则返回空串。题目要求分割出$x$的出现次数要大于等于$1$，也就是说不能分出空串。我们可以将$x$称为“最大公约串”。

首先，如果其中一个为空则直接返回空串。如果$s+t\ne t+s$也直接返回空串。然后算一下两个串长度的最大公约数$l$，则$s$的长$l$的前缀即为所求。

算法正确性证明：
首先，如果$s+t\ne t+s$，那么肯定不会有最大公约串，这是显然的，所以算法正确。设$s+t=t+s$，并且两串的长度的最大公约数等于$l$，如果两个串长度相等，那显然成立；否则的话，不妨设$s$短，如果$l_s|l_t$，容易证明成立。接着，我们只需要证明$s[0:l-1]$是最大公约串即可。为了方便起见，规定$s_{0l}=s[0:l-1],s_{1l}=s[l:2l-1],...$，对$t$也做类似规定。证明思路是利用字符串哈希。设$l=\gcd (l_s,l_t)$，并且$l_s=ul$，$l_t=vl$（即有$\gcd (u,v)=1$），令$s_{kl}$的哈希值表示为$s_k$，$t_{kl}$的哈希值表示为$t_k$，哈希用的素数要取的足够大，以至于两个哈希值相等就等价于字符串相等。考虑整系数多项式环$\mathbb{Z}[X]$，那么$s+t=t+s$等价于：$$\begin{gathered} s_0+s_{1}X+s_2{X}^2+...+s_{u-1}{X}^{u-1}+ \\ {X}^u(t_0+t_{1}X+...+t_{v-1}{X}^{v-1})= \\ {t}_0+t_{1}X+t_2{X}^2+...+t_{v-1}{X}^{v-1}+ \\ {X}^v(s_0+s_{1}X+...+s_{u-1}{X}^{u-1}) \end{gathered}$$这个式子说白了就是对应的长度$l$的片段相等。上面的式子等价于：$$\begin{gathered} (X^u-1)(t_0+t_{1}X+t_2{X}^2+...+t_{v-1}{X}^{v-1})= \\ (X^v-1)(s_0+s_{1}X+...+s_{u-1}{X}^{u-1}) \end{gathered}$$两边同时除以$X-1$得：$$\begin{gathered} (1+X+...+X^{u-1})(t_0+t_{1}X+t_2{X}^2+...+t_{v-1}{X}^{v-1}) \\ =(1+X+...+X^{v-1})(s_0+s_{1}X+...+s_{u-1}{X}^{u-1}) \end{gathered}$$比较常数项得$s_0=t_0$。由于$\gcd (u,v)=1$，所以$1+X+...+X^{u-1}$和$1+X+...+X^{v-1}$也互素（因为复数域上它们的根分别是$u$次单位根和$v$次单位根除去$1$，当$u$和$v$互素的时候，它们在$\mathbb{C}[X]$上是互素的，在$\mathbb{Z}[X]$上更互素）所以有$$(1+X+...+X^{u-1})|(s_0+s_{1}X+...+s_{u-1}{X}^{u-1})$$而这两个多项式次数相等，所以只能相差一个常数，这个常数只能是$s_0$，所以得到$s_0=s_1=...=s_{u-1}$，同理得$t_0=t_1=...=t_{v-1}$。又$s_0=t_0$，所以就有：$$s_0=s_1=...=s_{u-1}=t_0=t_1=...=t_{v-1}$$也就是说$s_0$就是最大公约串，结论成立。

代码如下：

class Solution {
 public:
  string gcdOfStrings(string s1, string s2) {
    if (s1 + s2 != s2 + s1) return "";
    return s1.substr(0, gcd(s1.size(), s2.size()));
  }

  int gcd(int a, int b) {
    return b ? gcd(b, a % b) : a;
  }
};

时间复杂度$O(l_s+l_t)$，空间$O(1)$。

注解：
字符串哈希来证明字符串的结论，可以将字符串问题转化为代数问题，问题的描述和解决都会更加清楚。