将一个给定字符串 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);
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zigzag-conversion
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:一开始没看懂Z是啥,结果是N字,然后就是N字排序,0和numRows-1位置是确定的,从每一行来看,其他位置都是距离某位置(i),下一个位置2 * numRows - 1 - i 个单位长度,再下一个位置又距离这个位置2*i,依次相互交替。
public class EverySeventeen {
public static void main(String[] args) {
System.out.println(convert("KarryWang",3));
}
public static String convert(String s, int numRows) {
if (numRows == 1) {//只有一个字符的特殊情况
return s;
}
StringBuilder sb = new StringBuilder();
for (int i = 0; i < numRows && i < s.length(); i++) {
int pos = i;
sb.append(s.charAt(pos));
for (int j = 0; ; j++) {
if (i % (numRows - 1) == 0) {
pos = pos + (numRows - 1) * 2;
} else {
pos = pos + (numRows - 1 - i) * 2 * ((j + 1) % 2) + i * 2 * (j % 2); //交替加两倍的 numRows-1-i 和 i
}
if (pos > s.length() - 1) {
break;
}
sb.append(s.charAt(pos));
}
}
return sb.toString();
}
}