竖直打印单词

这是我参与「掘金日新计划 · 2 月更文挑战」的第 10 天，点击查看活动详情

问题描述

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

思路分析

首先我们先要理解一下题目意思，题目会给我们一个字符串s，我们要按照单词在 s 中的出现顺序将它们全部竖直返回。如："HOW ARE YOU"

如上图：我们可以将字符串通过空格分隔开来得到三个单词，分别是 HOW、ARE、YOU，变成竖直的方法是将每个单词同一位置的字母提出拼到一起，如上面字符串得到的三个单词，三个单词的第一个字母分别是 H、A、Y，所以我们可以得到第一个竖直连接的字符串HAY，其他位置的字母也是同理。题目中还说到单词应该以字符串列表的形式返回，必要时用空格补位，但输出尾部的空格需要删除（不允许尾随空格），那么在什么情况下需要使用空格补位呢？如以下这个字符串：I AM FIND 如上图，将这个字符串的单词竖直打印，我们可以得到：["IAF"," MI"," N"," D"]，这里的" MI"," N"," D"都是需要空格进行补位，需要补位的空格的数量我们可以这样计算：

需要补位的空格数量 = 当前单词的下标 - 当前位置竖直单词长度

完整 AC 代码如下：

AC 代码

/**
 * @param {string} s
 * @return {string[]}
 */
var printVertically = function(s) {
    s = s.split(' ');
    const res = [];
    for(let i = 0; i < s.length; i++){
        for(let j = 0; j < s[i].length; j++){
            if(res.length < j + 1) res.push('');
            res[j] += ' '.repeat(Math.max(i - res[j].length,0)) + s[i][j];
        }
    }
    return res;
};

说在后面

本人为算法业余爱好者，平时只是随着兴趣偶尔刷刷题，如果上面分享有错误的地方，欢迎指出，感激不尽。