题目

给你一个字符串 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 仅含大写英文字母。
• 题目数据保证两个单词之间只有一个空格。

思路

一列一列的取

解法

class Solution {

    public List<String> printVertically(String s) {
        List<String> res = new ArrayList<>();
        String[] split = s.split(" ");
        int col = split[0].length();
        for (String str : split) {
            col = Math.max(col, str.length());
        }
        for (int c = 0; c < col; c++) {
            StringBuilder sb = new StringBuilder();
            for (String str : split) {
                sb.append(c < str.length() ? str.charAt(c) : " ");
            }
            res.add(trim(sb.toString()));
        }
        return res;
    }

    private String trim(String s) {
        int i = s.length() - 1;
        char[] chars = s.toCharArray();
        while (chars[i] == ' ') {
            i--;
        }
        return String.valueOf(chars, 0, i + 1);
    }
}

总结

• 分析出几种情况，然后分别对各个情况实现