题目
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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
链接:https://leetcode-cn.com/problems/zigzag-conversion
题解:
- 土鳖解法
思路:二位数组实现,然后按行进行拼接(注意判断字符是否为默认值,charArray[i][j] != 0)。
public class Solution {
public String convert(String s, int numRows) {
if (numRows == 1) {
return s;
}
String ans = "";
char[][] charArray = getCharArray(s, numRows);
for (int i = 0; i < numRows; i++) {
for (int j = 0; j < ((s.length() / (2 * numRows - 2)) + 1) * (numRows - 1); j++) {
if (charArray[i][j] != 0) {
ans = ans + String.valueOf(charArray[i][j]);
}
}
}
return ans;
}
char[][] getCharArray(String s, int numRows) {
int length = s.length();
int leftNumber = length % (2 * numRows - 2);
int count = length / (2 * numRows - 2);
char[][] charArray = new char[numRows][(count + 1) * (numRows - 1)];
int index = 0;
for (int i = 0; i < count; i++) {
for (int j = 0; j < numRows; j++) {
charArray[j][i * (numRows - 1)] = s.charAt(index++);
}
for (int k = 1; k < numRows - 1; k++) {
charArray[numRows - k - 1][i * (numRows - 1) + k] = s.charAt(index++);
}
}
// System.out.println(Arrays.deepToString(charArray));
if (leftNumber > numRows) {
for (int j = 0; j < numRows; j++) {
charArray[j][count * (numRows - 1)] = s.charAt(index++);
}
for (int m = 1; m <= leftNumber - numRows; m++) {
charArray[numRows - m - 1][count * (numRows - 1) + m] = s.charAt(index++);
}
} else {
for (int j = 0; j < leftNumber; j++) {
charArray[j][count * (numRows - 1)] = s.charAt(index++);
}
}
return charArray;
}
}
- 按行排序
思路:添加min(numRows,len(s)) 个非空行,按字符串s的字符顺序给行添加,同时设置标志位,遇到0,需要向下一行;遇到numRows - 1,需要向上一行,所有行添加完毕后,进行拼接;
class Solution {
public String convert(String s, int numRows) {
if (numRows <= 1) {
return s;
}
String ans = "";
List<StringBuilder> row = new ArrayList<>();
for (int i = 0; i < Math.min(numRows, s.length()); i++) {
row.add(new StringBuilder());
}
int curRow = 0;
boolean goingDown = false;
for (char c : s.toCharArray()) {
row.get(curRow).append(c);
if (curRow == 0 || curRow == numRows - 1) {
goingDown = !goingDown;
}
curRow += goingDown ? +1 : -1;
}
for (StringBuilder sb: row) {
ans += sb.toString();
}
return ans;
}
}
- 按行访问
思路:通过规律找到每一行的数字在字符串中的位置,再逐行进行拼接。
代码待补充。