这题以前写过,重新比第一次写快很多,思路很简单,看来刷题还是有用的!此次代码比以前的代码更加精炼。
The string "PAYPALISHIRING"
is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)
P A H N
A P L S I I G
Y I R
And then read line by line: "PAHNAPLSIIGYIR"
Write the code that will take a string and make this conversion given a number of rows:
string convert(string s, int numRows);
Example 1:
Input: s = "PAYPALISHIRING", numRows = 3
Output: "PAHNAPLSIIGYIR"
Example 2:
Input: s = "PAYPALISHIRING", numRows = 4
Output: "PINALSIGYAHRPI"
Explanation:
P I N
A L S I G
Y A H R
P I
solution:(时间复杂度为O(n),由图形我们不难发现一个循环circle是(numRows-1)*2, 在第0行和第numRows-1行,每个循环都是只有一个数,而对于这两行中间的数据,除了最后一个循环,一行必然两个数据,且两数index之差为(circle-row)-row)
public class Solution006 {
public String convert(String s, int numRows) {
if (numRows==1)
return s;
int circle = (numRows-1)*2;
int len = s.length();
char [] ans = new char [len];
int k=0;
for(int row=0;row < numRows;row++){
for(int i=row;i<len;i+=circle){
ans[k++] =s.charAt(i);
if(row !=0 &&row!=numRows-1 && i+ circle-row*2<len)
ans[k++]=s.charAt(i+ circle-row*2);
/*
为了解释i+ circle-row*2
对于每一轮循环,如下例,对每一轮循环大小为6的数字
编号0,1,2,3,4,5,
分别P,A,Y,P,A,L
可见1与5号在同行,2与4号在同行,其和都为6,所以可以得到两者间的距离为
(circle-row)-row
P I N
A L S I G
Y A H R
P I
*/
}
}
return new String(ans);
}
}