力扣2138-将字符串拆分为若干长度为 k 的组已解答简单

字符串 s 可以按下述步骤划分为若干长度为 k 的组:

  • 第一组由字符串中的前 k 个字符组成,第二组由接下来的 k 个字符串组成,依此类推。每个字符都能够成为 某一个 组的一部分。
  • 对于最后一组,如果字符串剩下的字符 不足 k 个,需使用字符 fill 来补全这一组字符。

注意,在去除最后一个组的填充字符 fill(如果存在的话)并按顺序连接所有的组后,所得到的字符串应该是 s 。

给你一个字符串 s ,以及每组的长度 k 和一个用于填充的字符 fill ,按上述步骤处理之后,返回一个字符串数组,该数组表示 s 分组后 每个组的组成情况 。

示例 1:

输入:s = "abcdefghi", k = 3, fill = "x"
输出:["abc","def","ghi"]
解释:
前 3 个字符是 "abc" ,形成第一组。
接下来 3 个字符是 "def" ,形成第二组。
最后 3 个字符是 "ghi" ,形成第三组。
由于所有组都可以由字符串中的字符完全填充,所以不需要使用填充字符。
因此,形成 3 组,分别是 "abc"、"def" 和 "ghi" 。

示例 2:

输入:s = "abcdefghij", k = 3, fill = "x"
输出:["abc","def","ghi","jxx"]
解释:
与前一个例子类似,形成前三组 "abc"、"def" 和 "ghi" 。
对于最后一组,字符串中仅剩下字符 'j' 可以用。为了补全这一组,使用填充字符 'x' 两次。
因此,形成 4 组,分别是 "abc"、"def"、"ghi" 和 "jxx" 。
from typing import List

class Solution:
    def divideString(self, s: str, k: int, fill: str) -> List[str]:
        # 初始化结果列表
        result = []
        
        # 遍历字符串,按 k 的长度分组
        for i in range(0, len(s), k):
            group = s[i:i+k]
            # 如果组的长度不足 k,则用填充字符补全
            if len(group) < k:
                group += fill * (k - len(group))
            result.append(group)
        
        return result

解释:

  1. 使用 for 循环遍历字符串 s,每次截取 k 个字符。如果最后一组字符不足 k,则用填充字符 fill 补足。
  2. 将每一组加入结果列表并返回。

结果:

对于输入 s = "abcdefghi", k = 3, fill = "x",输出为 ['abc', 'def', 'ghi']

对于输入 s = "abcdefghij", k = 3, fill = "x",输出为 ['abc', 'def', 'ghi', 'jxx']

 可以通过减少不必要的判断和操作对代码进行优化。具体来说,可以在开始时直接判断是否需要补全字符串,再进行分组,从而避免在循环中多次检查长度不足的情况。

优化后的代码如下:

from typing import List

class Solution:
    def divideString(self, s: str, k: int, fill: str) -> List[str]:
        # 计算需要补充的字符数量
        remainder = len(s) % k
        if remainder != 0:
            s += fill * (k - remainder)  # 用填充字符补全字符串
        
        # 按 k 的长度进行分组
        return [s[i:i+k] for i in range(0, len(s), k)]

 优化点:

  1. 提前处理填充:在循环之前先判断字符串是否需要填充字符,这样可以避免在每次循环中都进行判断。
  2. 列表推导式:使用列表推导式代替循环生成分组,代码更加简洁且高效。
     

    优化效果:

    通过减少每次分组时的判断,代码运行更加高效,同时提高了可读性。

猜你喜欢

转载自blog.csdn.net/qq_73454087/article/details/142288363