6. Z 字形变换
题目描述
将一个给定字符串 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”
解题思路
用一个numRows行N列的字符串数组,按规律地存成z字型。存的顺序为:先列,后斜对角。因此,一列和一斜线构成一组,用循环的方式存完即可。存好之后,再逐行打印即可。代码如下:
class Solution {
public static String convert(String s, int numRows) {
char arr [][] = new char[numRows][1001];
int k = 0;
int i = 0;
int j = 0;
while (k < s.length()){
//写列
for(int l = 0; l < numRows; l++){
if (k >= s.length()){
break;
}
arr[i][j] = s.charAt(k);
k++;
i++;
}
//写对角
i = numRows - 2;
j++;
for (int l = 0; l < numRows - 2; l++){
if (k >= s.length()){
break;
}
arr[i][j] = s.charAt(k);
k++;
i--;
j++;
}
i = 0;
}
StringBuilder sb = new StringBuilder();
for (int l = 0; l < arr.length; l++) {
for (int m = 0; m < arr[0].length; m++) {
if (arr[l][m] == 0){
;//System.out.print(" ");
}else {
// System.out.print(arr[l][m]);
sb.append(arr[l][m]);
}
}
// System.out.println();
}
String str = sb.toString();
// System.out.println(sb.toString());
return str;
}
}