1324. 竖直打印单词

清疚 2023-10-04 09:34 1阅读 0赞

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

package Solution1324;
    
    import java.util.ArrayList;
    import java.util.List;
    
    class Solution {
    	public List<String> printVertically(String s) {
    		List<String> out = new ArrayList<String>();
    		int max = 0;
    		String[] arr = s.split("\\s+");
    		for (int i = 0; i < arr.length; i++) {
    			if (arr[i].length() > max) {
    				max = arr[i].length();
    			}
    		}
    		for (int i = 0; i < max; i++) {
    			String temp = "";
    			for (int j = 0; j < arr.length; j++) {
    				if (arr[j].length() > i) {
    					temp = temp + arr[j].substring(i, i + 1);
    				} else {
    					temp = temp + " ";
    				}
    			}
    			temp = temp.replaceAll("\\s+$", "");
    			out.add(temp);
    		}
    		return out;
    	}
    
    	public static void main(String[] args) {
    
    		Solution sol = new Solution();
    
    		String s = "TO BE OR NOT TO BE";
    
    		System.out.println(sol.printVertically(s));
    	}
    
    }