LeetCode-Python-1163. 按字典序排在最后的子串

给你一个字符串 s,找出它的所有子串并按字典序排列,返回排在最后的那个子串。

 

示例 1:

输入:"abab"
输出:"bab"
解释:我们可以找出 7 个子串 ["a", "ab", "aba", "abab", "b", "ba", "bab"]。按字典序排在最后的子串是 "bab"。

示例 2:

输入:"leetcode"
输出:"tcode"

 

提示:

  1. 1 <= s.length <= 10^5
  2. s 仅含有小写英文字符。

第一种思路:

双重循环 + 排序的暴力解是非常好想的,但是本题要求O(N)所以不行……

第二种思路:

思考一下可以发现,一个字符串字典序最大的子串的第一个字母一定是 组成这个字符串的字典序最大的字母。

比如 abbbed 里,字典序最大的字母为e,那么可以肯定,这个字符串的字典序最大的子串一定是e打头。

所以只要找到输入字符串里的字典序最大的字母,然后把以这个字母开头的所有子串找出来,从中找最大的即可。

试验了一下发现出现了新的问题,上述算法在处理 “a” * 100000 这个输入的时候会超时,

于是继续考虑优化,

假设有输入为 zzzabd,根据上述思想,应该可以找到 zzzabb, zzabb 和 zabb三个子串,

此时可以这么想,如果当前子串的前一个字母也是 字典序最大的字母,那么当前子串必不可能是答案,

因为 zzzabb 比 zzabb 多一个z,所以前者的字典序肯定 > 后者。

因此可以利用这个条件加快查找速度。

class Solution(object):
    def lastSubstring(self, s):
        """
        :type s: str
        :rtype: str
        """
        alphabet = "abcdefghijklmnopqrstuvwxyz"[::-1]
        res = ""
        flag = 0
        for char in alphabet: #按字典序从大到小找
            for i in range(len(s)):
                if i and char == s[i - 1]: #必备优化
                    continue
                if char == s[i]:
                    flag = 1
                    res = max(res, s[i:])
            if flag: #如果已经在当前的字母里找到结果,那么不用看字典序更小的字母了
                break
            
        return res

 

你可能感兴趣的:(Leetcode,字符串)