题目:
将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列。比如输入字符串为 "PAYPALISHIRING" 行数为 3 时,排列如下:
P A H N
A P L S I I G
Y I R
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如: "PAHNAPLSIIGYIR"
。
请你实现这个将字符串进行指定行数变换的函数:
string convert(string s, int numRows);
示例1:
输入:s = "PAYPALISHIRING", numRows = 3
输出:"PAHNAPLSIIGYIR"
示例2:
输入:s = "PAYPALISHIRING", numRows = 4
输出:"PINALSIGYAHRPI"
解释:
P I N
A L S I G
Y A H R
P I
示例3:
输入:s = "A", numRows = 1
输出:"A"
提示:
1 <= s.length <= 1000
s
由英文字母(小写和大写)、','
和'.'
组成1 <= numRows <= 1000
解题:
其实 N 字形是这样的
- 可以根据字母排列的方式,找到每行字母变化的规律;
- 按行进行寻找,一共要遍历 numRows 次字符串;
- 需要设置一个 Boolean变量,代表当前对字符串扫描的方向;
- 当遍历到某一行时,如果是当前对字符串扫描的方向是向下的,则该行下一个字母的下标是 j + (2*(numRows-i)) - 2,j 代表当前字母的下标,i 代表的是当前的行数,当找到下一个字母之后,需要转换扫表的方向;
- 如果是当前对字符串扫描的方向是向上的,则该行下一个字母的下标是 j + (2*(numRows-i)) - 2,j 代表当前字母的下标,i 代表的是当前的行数,当找到下一个字母之后,需要转换扫表的方向;
- 对于第一行和最后一行,其实不需要转换扫描的方向,但是为了和我们找到的规律相同,则加一个特殊判断,只有当找到的下一个字母的下标和当前的下标不一样时,才将找到的下一个字母加入到 convertStr 中;
- 还有一个特殊情况就是,当nu时,需要做特殊判断,不然会陷入死循环。
class Solution {
public String convert(String s, int numRows) {
String convertStr = "";
boolean flag;
if(numRows == 1) return s;
for (int i = 0; i < numRows; i++){
flag = false;
for (int j = i; j < s.length();){
if (!flag){
if (j != j + (2*(numRows-i)) - 2){
convertStr += s.charAt(j);
j = j + (2*(numRows-i)) - 2;
}
}else {
if (j != j + (2*(i+1)) - 2){
convertStr += s.charAt(j);
j = j + (2*(i+1)) - 2;
}
}
flag = !flag;
}
}
return convertStr;
}
}