【题目描述】
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时,排列如下:
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:“LCIRETOESIIGEDHN”。
请你实现这个将字符串进行指定行数变换的函数:
string convert(string s, int numRows);
【解题思路】
对于Z字形,每行使用一个StringBuiler来记录,所以得到一个数组StringBuiler[numRows];通过数组下标y,记录当前方向的up,开始模拟构造Z字型。
【代码】
String convert(String s, int numRows) {
if (numRows == 1) {
return s;
}
StringBuilder[] builders = new StringBuilder[numRows];
for (int i = 0; i < numRows; i++) {
builders[i] = new StringBuilder();
}
// 核心代码快
char[] arr = s.toCharArray();
int y = 0;
boolean up = false;
for (int i = 0; i < arr.length; i++) {
builders[y].append(arr[i]);
if (up) { // 自下向上
y--;
if (y == -1) {
y = 1;
up = false;
}
} else { // 自上向下
y++;
if (y == numRows) {
y = numRows - 2;
up = true;
}
}
}
StringBuilder result = new StringBuilder(numRows);
for (StringBuilder b: builders) {
result.append(b.toString());
}
return result.toString();
}
【运行结果】
执行用时 : 6 ms, 在所有 java 提交中击败了85.36%的用户;
内存消耗 : 39 MB, 在所有 java 提交中击败了93.94%的用户。
【小结】
算法的执行效率有待提高,但较易于理解。大家如有好的看法,欢迎补充。