java 竖直打印单词_leetcode 1324: 竖直打印单词

import java.util.ArrayList;

import java.util.List;

/**

* @Class PrintVertically

* @Description 1324. 竖直打印单词

* 给你一个字符串 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 仅含大写英文字母。

* 题目数据保证两个单词之间只有一个空格。

* @Author

* @Date 2020/6/23

**/

public class PrintVertically {

}

public static List printVertically(String s) {

if (s == null) {

return null;

}

String[] strings = s.split(" ");

int maxlen = 0;

for (int i = 0; i < strings.length; i++) {

maxlen = Math.max(maxlen, strings[i].length());

}

List stringList = new ArrayList<>();

for (int j = 0; j < maxlen; j++) {

// 保存竖直排列后的一行

StringBuilder temp = new StringBuilder("");

// blankTemp保存每行大写字符之间的空格，最后一个大写字符后的空格串直接丢弃

StringBuilder blankTemp = new StringBuilder("");

for (int i = 0; i < strings.length; i++) {

if (j >= strings[i].length()) {

blankTemp.append(" ");

} else {

temp.append(blankTemp);

blankTemp = new StringBuilder("");

temp.append(strings[i].charAt(j));

}

}

stringList.add(temp.toString());

}

return stringList;

}

// 测试用例

public static void main(String[] args) {

String s = "HOW ARE YOU";

List ans = printVertically(s);

System.out.print("demo01 result:[");

for (int i = 0; i < ans.size(); i++) {

System.out.print("," + ans.get(i));

}

System.out.println("]");

s = "TO BE OR NOT TO BE";

ans = printVertically(s);

System.out.print("demo02 result:[");

for (int i = 0; i < ans.size(); i++) {

System.out.print("," + ans.get(i));

}

System.out.println("]");

s = "CONTEST IS COMING";

ans = printVertically(s);

System.out.print("demo03 result:[");

for (int i = 0; i < ans.size(); i++) {

System.out.print("," + ans.get(i));

}

System.out.println("]");

}