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