给你一个字符串 s
。请你按照单词在 s
中的出现顺序将它们全部竖直返回。
单词应该以字符串列表的形式返回,必要时用空格补位,但输出尾部的空格需要删除(不允许尾随空格)。
每个单词只能放在一列上,每一列中也只能有一个单词。
示例 1:
输入:s = "HOW ARE YOU"
输出:["HAY","ORO","WEU"]
解释:每个单词都应该竖直打印。
"HAY"
"ORO"
"WEU"
示例 2:
输入:s = "TO BE OR NOT TO BE"
输出:["TBONTB","OEROOE"," T"]
解释:题目允许使用空格补位,但不允许输出末尾出现空格。
"TBONTB"
"OEROOE"
" T"
示例 3:
输入:s = "CONTEST IS COMING"
输出:["CIC","OSO","N M","T I","E N","S G","T"]
提示:
1 <= s.length <= 200
s
仅含大写英文字母。- 题目数据保证两个单词之间只有一个空格。
解题思路
按照题目意思操作,首先将字符串通过空格分开得到字符串数组,然后统计该数组中最长的字符串长度max_len
(也就是最后返回数组的大小)。接着就是将每一列字符添加到结果中去,过程中需要判断当前遍历到的字符串长度是不是小于当前遍历的列数。如果小于的话,那么添加空格;否则,添加字符串对应列的字符即可。
class Solution:
def printVertically(self, s: str) -> List[str]:
strs = s.split(" ")
max_len = max([len(st) for st in strs])
res = [""] * max_len
for j in range(max_len):
for st in strs:
if len(st) > j:
res[j] += st[j]
else:
res[j] += " "
for i in range(max_len):
res[i] = res[i].rstrip(' ')
return res
pythonic
的写法:
class Solution:
def printVertically(self, s: str) -> List[str]:
return [''.join(a).rstrip() for a in itertools.zip_longest(*s.split(), fillvalue=' ')]
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!