【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中所有的字符都写完。

代码如下：

 public class Solution {
     public String convert(String s, int nRows) {
         if(s.length() <= nRows || nRows == 1)
             return s;
         StringBuffer[] sB = new StringBuffer[nRows];
         for(int i = 0;i < nRows;i++)
             sB[i] = new StringBuffer();

         int kepeler = 0;
         boolean add = true;
         for(int i=0;i < s.length();i++){
             sB[kepeler].append(s.charAt(i));
             if(kepeler == 0)
                 add = true;
             if(kepeler == nRows-1)
                 add = false;
             if(add)
                 kepeler++;
             else
                 kepeler--;
         }
         String answer = new String();
         for(StringBuffer temp:sB)
             answer += temp.toString();

         return answer;
     }
 }

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