【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 的错误。