leetcode_1324 竖直打印单词

248 阅读1分钟

要求

给你一个字符串 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"]

核心代码

class Solution:
    def printVertically(self, s: str) -> List[str]:
        l = s.split(" ")
        max_l = max([len(item) for item in l])
        res = [""] * max_l
        for i in range(max_l):
            tmp = ""
            for j in range(len(l)):
                if len(l[j]) > i:
                    tmp += l[j][i]
                else:
                    tmp += " "
            res[i] = tmp.rstrip()
        return res

image.png

解题思路:就是按照题意完成单词的拼接功能,比较简单,注意不存在的时候用" "进行补齐即可,最后按照题意取出右侧的多余空格。