个人题解与官方题解二思路相同,但官方题解的代码表达更简洁。
题目
链接:https://leetcode-cn.com/problems/zigzag-conversion
官方题解一:按行排序
思路
通过从左向右迭代字符串,我们可以轻松地确定字符位于 Z 字形图案中的哪一行。
算法
我们可以使用 min(numRows,len(s))个列表来表示 Z 字形图案中的非空行。
从左到右迭代 s,将每个字符添加到合适的行。可以使用当前行和当前方向这两个变量对合适的行进行跟踪。
只有当我们向上移动到最上面的行或向下移动到最下面的行时,当前方向才会发生改变。
class Solution {
public String convert(String s, int numRows) {
if (numRows == 1) return s;
List<StringBuilder> 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)
作者:LeetCode
链接:https://leetcode-cn.com/problems/zigzag-conversion/solution/z-zi-xing-bian-huan-by-leetcode/
来源:力扣(LeetCode)
代码(带注释)
import java.util.ArrayList;
import java.util.List;
/**
* leetcode-6-middle
*/
public class zigzag_conversion {
static class Solution {
/**
* 个人题解
* 思路:1、同一行的竖列上的字符之间间隔和斜列上的字符之间的间隔均为gap,依次拼接成结果字符串;
* 2、第一行和最后一行只有竖列没有斜列
*/
String convert(String s, int numRows) {
if(numRows >= s.length() || numRows == 1) return s;
int gap = 2 * numRows - 2; //每列两个对应元素的间隔
StringBuilder ans = new StringBuilder(); //存储结果值的StringBuilder对象
for(int i = 1; i <= numRows; i++){ //对每行进行遍历
int m = i; //java中每次循环不会重置i的值,所以内部对i的值进行修改的话会影响循环控制
if( m == 1 || m == numRows){ //第一行和最后一行只需根据第一个元素的位置依次累加gap即可
while (m <= s.length()) {
ans.append(s.charAt(--m));
m += gap + 1;//因为检索时减去了1所以需要加回来
}
continue;
}
int j = 2 * numRows - m; // j用来表示处于两竖列之间的元素的位置
while(m <= s.length() || j <= s.length()){
if(m <= s.length()) {
ans.append(s.charAt(--m));
m += gap + 1;
}
if(j <= s.length()) {
ans.append(s.charAt(--j));
j += gap + 1;
}
}
}
return ans.toString();
}
/**
* 官方题解二,思路与上同,但表达更简洁
* @param s:需要转换的字符串
* @param numRows : 指定Z字形的最大行数
* @return 返回Z字形转换后按行重组的字符串
*/
String convertBetter(String s, int numRows) {
if (numRows == 1) return s;
StringBuilder ret = new StringBuilder();
int n = s.length();
int cycleLen = 2 * numRows - 2; //即gap
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)); //根据竖列上的字符与同行相邻的斜列上的字符的数量关系,得出j的位置
}
}
return ret.toString();
}
/**
* 官方题解一
* 思路: 沿着Z字形依次遍历,将每个字符放入其行对应的StringBuilder对象中
*/
String convertA(String s, int numRows) {
if (numRows == 1 || numRows >= s.length()) return s;
List<StringBuilder> rows = new ArrayList<>(); //存储用以顺序每行字符的StringBuilder对象
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();
}
}
}