将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “PAYPALISHIRING” 行数为 3 时,排列如下:
P A H N
A P L S I I G
Y I R
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:“PAHNAPLSIIGYIR”。
public class Problems4 {
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner scanner = new Scanner(System.in);
String input = scanner.nextLine();
int lineNum = scanner.nextInt();
String result = printString(input, lineNum);
System.out.println(result);
}
private static String printString(String input, int lineNum) {
// TODO Auto-generated method stub
HashMap<Integer, StringBuilder> map = new HashMap<>();
if (lineNum <= 1) {
return input;
}
for (int i = 0; i < input.length(); i++) {
int temp = i % (2 * lineNum - 2);
if ((temp + 1) / lineNum > 0) {
temp = (lineNum - 1 - ((temp + 1) % lineNum));
}
StringBuilder sb = map.get(temp);
if (sb == null) {
sb = new StringBuilder();
}
map.put(temp, sb.append(input.charAt(i)));
}
StringBuilder builder = new StringBuilder();
for (int i = 0; i < map.size(); i++) {
builder.append(map.get(i));
}
return builder.toString();
}
}