【leetcode刷题笔记】ZigZag Conversion

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" .

开一个StringBuffer的数组，共nRows行，然后按照0->nRows-1->0->nRows-1->......的顺序填写这个数组，用指针kepeler表示将当前的字符加到kepeler行的StringBuffer的结尾，kepeler从0递增到nRows-1，再从nRows-1递减到0，再从0递增到nRows-1......用一个变量add标志kepeler当前是要递增还是递减，当kepeler递增到nRows-1或者递减到0的时候改变add的真假。以题目中的例子来说，kepeler的值变化如下：0->1->2->1->0->1->2->......直到把s中所有的字符都写完。

代码如下：

 1 public class Solution {
 2     public String convert(String s, int nRows) {
 3         if(s.length() <= nRows || nRows == 1)
 4             return s;
 5         StringBuffer[] sB = new StringBuffer[nRows];
 6         for(int i = 0;i < nRows;i++)
 7             sB[i] = new StringBuffer();
 8
 9         int kepeler = 0;
10         boolean add = true;
11         for(int i=0;i < s.length();i++){
12             sB[kepeler].append(s.charAt(i));
13             if(kepeler == 0)
14                 add = true;
15             if(kepeler == nRows-1)
16                 add = false;
17             if(add)
18                 kepeler++;
19             else
20                 kepeler--;
21         }
22         String answer = new String();
23         for(StringBuffer temp:sB)
24             answer += temp.toString();
25
26         return answer;
27     }
28 }

要注意的一点就是代码中第5行虽然申请了StringBuffer的数组，但是并没有为数组中每个元素分配空间，所以要用一个循环来分配空间，否则会报 java.lang.NullPointerException 的错误。

【leetcode刷题笔记】ZigZag Conversion