将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码:`class Solution {
public String convert(String s, int numRows) {
int n=s.length();
if(n==1)
return s;
char[][] ls= new char[1000][1000];
int[] num=new int[numRows];
for(int k=0;k<numRows;k++)
num[k]=0;
int i=0,local=0;
boolean down=true;
while(i<n){
if(down){
ls[local][num[local]]=s.charAt(i);
num[local]=num[local]+1;
local++;
if(local>=numRows){//需要改变方向
local=numRows-2;
down=false;
if(numRows==1)
local=0;
}
}
else{
ls[local][num[local]]=s.charAt(i);
num[local]++;
local--;
if(local<0){//第一行放入了字符需要改变方向
local=1;
down=true;
if(numRows == 1)
local=0;
}
}
i++;
}
String str="";
int m=0;
for(i=0;i<numRows;i++){
for(int j=0;j<num[i];j++)
str=str+ls[i][j];
}
return str;
}
}`
思路:因为行数是已知的,我们并不需要将字符放为Z字形,只要字符在其对应的行上就ok,eg:
L D R
E O E I I
E C I H N 此为Z字形
T S G
LDR
EOE II
ECIHN 我们只需要对应字符在对应的行上即可
TS G
而因为行数的已知所以我们只需要一个位置变量local来控制字符应该放在那一行即可,我们创建一个数组,只使用数组的numRows行,给每一行一个记录长度的值,然后控制local的位置来放入字符即可;