Java从左到右进行w字形排序_Java算法练习—— Z 字形变换

题目描述

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，排列如下：

L C I R

E T O E S I I G

E D H N

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："LCIRETOESIIGEDHN"。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例 1

输入: s = "LEETCODEISHIRING", numRows = 3

输出: "LCIRETOESIIGEDHN"

示例 2

输入: s = "LEETCODEISHIRING", numRows = 4

输出: "LDREOEIIECIHNTSG"

解释:

L D R

E O E I I

E C I H N

T S G

题解 1 (按行排序)

public String convert(String s, int numRows) {

if (numRows == 1) return s;

List rows = new ArrayList<>();

for (int i = 0; i < Math.min(numRows, s.length()); i++)

rows.add(new StringBuilder());

int curRow = 0;

boolean goingDown = false;

for (char c : s.toCharArray()) {

rows.get(curRow).append(c); // 按行存值

if (curRow == 0 || curRow == numRows - 1) goingDown = !goingDown; // 转向,很独特的方法,学到了

curRow += goingDown ? 1 : -1; // 递增或递减

}

StringBuilder ret = new StringBuilder();

for (StringBuilder row : rows) ret.append(row); // 拼接字符串

return ret.toString();

}

复杂度分析

时间复杂度：$O(n)$，其中$ n==len(s)$

空间复杂度：$O(n)$

题解 2 (按行访问)

public String convert(String s, int numRows) {

if (numRows == 1) return s;

StringBuilder ret = new StringBuilder();

int n = s.length();

int cycleLen = 2 * numRows - 2; // 一个循环的长度

for (int i = 0; i < numRows; i++) {

for (int j = 0; j + i < n; j += cycleLen) {

ret.append(s.charAt(j + i));

if (i != 0 && i != numRows - 1 && j + cycleLen - i < n)

ret.append(s.charAt(j + cycleLen - i));

}

}

return ret.toString();

}

复杂度分析

时间复杂度：$O(n)$，其中 $n==len(s)$。每个索引被访问一次。

空间复杂度：$O(n)$。

手记

两种方法都不错，尤其是第一种。 可惜我个菜鸡，自己写了半天超内存是最难受的。