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

题意解析：

这道题很有趣，意思就是给你个字符串，按照锯齿状的样子写（见上方示例），写的时候竖着写，要返回写完以后横着读的结果。

解题思路：

既然要求是横着读，那我们就横着存储就行了。根据行数来定义数组，每个数组保存当前行的字符。代码如下：

public class Solution {
    public String convert(String s, int numRows) {
        char ss[] = s.toCharArray();
        String[] res = new String[numRows];
        //初始化String防止为null
        for(int i = 0; i < res.length; i++){
            res[i] = "";
        }
        for(int i = 0; i < ss.length;){
            for(int a = 0; a < numRows && i < ss.length; a++){
                //竖直下降
                res[a] += ss[i++];
            }
            for(int b = numRows - 2; b > 0 && i < ss.length; b--){
                res[b] += ss[i++];
            }
        }
        String result = "";
        for(int i = 0; i < numRows; i++){
            result += res[i];
        }
        return result;
    }
}

不过其实用String效率是有点低的，如果你对Java了解比较深的话，其实还有更好的办法，代码如下：

public class Solution {
    public String convert(String s, int nRows) {
        char[] c = s.toCharArray();
        int len = c.length;
        StringBuffer[] sb = new StringBuffer[nRows];
        for (int i = 0; i < sb.length; i++) sb[i] = new StringBuffer();
    
        int i = 0;
        while (i < len) {
            for (int idx = 0; idx < nRows && i < len; idx++) // vertically down
                sb[idx].append(c[i++]);
            for (int idx = nRows-2; idx >= 1 && i < len; idx--) // obliquely up
                sb[idx].append(c[i++]);
        }
        for (int idx = 1; idx < sb.length; idx++)
            sb[0].append(sb[idx]);
        return sb[0].toString();
    }
}

快了多少呢？我们可以来看一看

下图是第一份代码的速度：

下图是第二份代码的速度：

差异有点大。。所以建议大家还是使用StringBuffer把。