题目描述:
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zigzag-conversion
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
1.将字符先从上至下,再从下至上变形,再进行输出
2.可以设置标志位,从上至下标志位不断增加,当标志位到达最大行数时停止向下,并向上依次递减至第0行,反复迭代次过程直到所有字符输出。
题解
public String convert(String s, int numRows) {
if(numRows==1) return s;
List<StringBuilder> rowList= new ArrayList<>();
for(int i=0;i<Math.max(s.length(),numRows);i++) rowList.add(new StringBuilder());
/**
* 根据行数生成多个字符串序列
* 字符串序列的数目由行数和原始字符串长度共同决定。
* 1.如果字符串长度小于行数,那么生成字符串长度的字符序列,这是因为
* 字符串长度不够行数,每行将只有一个字符,那么生成字符串长度的字符串
* 序列就可以了。
* 2.字符串长度大于行数,代表每行大于1个字符,这时候按照行数生成字符
* 串序列。
*/
int i=0;
int flag=-1;
for(char c:s.toCharArray())
{
rowList.get(i).append(c);
if(i==0||numRows-i==1) flag=-flag;
/**
* 当行数等于0或最后一行时,我们需要改变行进方向
* 这个思路很精妙,考虑记住
*/
i+=flag;
}
StringBuilder res = new StringBuilder();
for(StringBuilder row : rowList) res.append(row);
return res.toString();
}
感悟
1.行进方向用flag标志位表示,这非常精妙。