字符串 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
解释:
- 使用
for
循环遍历字符串s
,每次截取k
个字符。如果最后一组字符不足k
,则用填充字符fill
补足。- 将每一组加入结果列表并返回。
结果:
对于输入
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)]
优化点:
- 提前处理填充:在循环之前先判断字符串是否需要填充字符,这样可以避免在每次循环中都进行判断。
- 列表推导式:使用列表推导式代替循环生成分组,代码更加简洁且高效。
优化效果:
通过减少每次分组时的判断,代码运行更加高效,同时提高了可读性。