leetcode_1071 字符串的最大公因子

312 阅读1分钟

要求

对于字符串 S 和 T,只有在 S = T + ... + T(T 自身连接 1 次或多次)时,我们才认定 “T 能除尽 S”。

返回最长字符串 X,要求满足 X 能除尽 str1 且 X 能除尽 str2。

示例 1:

输入:str1 = "ABCABC", str2 = "ABC"
输出:"ABC"

示例 2:

输入:str1 = "ABABAB", str2 = "ABAB"
输出:"AB"

示例 3:

输入:str1 = "LEET", str2 = "CODE"
输出:""

核心代码

class Solution:
    def gcdOfStrings(self, str1: str, str2: str) -> str:
        l1,l2 = len(str1),len(str2)
        set1,set2 = set(),set()
        for i in range(l1 + 1):
            if self.find(str1,str1[:i]):
                set1.add(str1[:i])
        
        for i in range(l2 + 1):
            if self.find(str2,str2[:i]):
                set2.add(str2[:i])
        
        res = list(set1 & set2)
        res = sorted(res,key = lambda x:-len(x))
        return res[0] if res else ""

    def find(self,string,tmp):
        string = string.replace(tmp,"")
        return len(string) == 0

image.png

解题思路:因为子串都是连续的,所以我们从第一个字符开始,将所有子串都遍历出来放到一个集合中,同时另一个操作也是一样的,最终得到两个集合的交集,我们排序,得到最长的那个即可。