将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时,排列如下:
那么将返回字符串LCIRETOESIIGEDHN;
如下为实现代码:(将其想象成一个二维数组)
public static String convert(String s, int numRows) {
//判断行数是否为1
if (numRows == 1)
return s;
//找出字符串长度和行数较小的中较小的
int len = (s.length() <= numRows) ? s.length() : numRows;
String[] rows = new String[len];
for (int i = 0; i < len; i++)
rows[i] = "";
int loc = 0;
boolean down = false;
for (int i = 0; i < s.length(); i++) {
rows[loc] += s.substring(i, i + 1);
if (loc == 0 || loc == numRows - 1)
down = !down;
loc += down ? 1 : -1;
}
String ans = "";
for (String row : rows) {
ans += row;
}
return ans;
}