题目:
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 text, int nRows);
convert("PAYPALISHIRING", 3)
should return "PAHNAPLSIIGYIR"
.
思路:
之字形变换,找数学方法。头尾两行,每个字符之间隔了2*nRows-2个,中间的行,每个字符都往下或往上加一个三角的长度。
规则:
//第0至第nRows-1行,下一跳 + 2*nRows-2;
//第1至nRows-2行(第i行),在偶数个角,下一跳 + (nRows-1-i)*2; 在奇数个角,下一跳+ i*2
public String convert(String s, int nRows) {
//规则:
//第0至第nRows-1行,下一跳 + 2*nRows-2;
//第1至nRows-2行(第i行),在偶数个角,下一跳 + (nRows-1-i)*2; 在奇数个角,下一跳+ i*2
int len = s.length();
StringBuffer sb = new StringBuffer();
if(nRows == 1 || nRows == len){
return s;
}
for(int i=0; i<nRows; i++){
int j=i, k=0;
while(j < len){
sb=sb.append(s.charAt(j));
if(i == 0 || i == nRows-1){
j+=(2*nRows-2);
} else{
if(k%2 == 0){//偶数个角
j+=2*(nRows-1-i);
} else{
j+=2*i;
}
k++;
}
}
}
return sb.toString();
}