【题目描述】
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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%的用户。
【小结】
算法的执行效率有待提高,但较易于理解。大家如有好的看法,欢迎补充。