LeetCode第六题 Z字形转换

将字符串 “PAYPALISHIRING” 以Z字形排列成给定的行数：（下面这样的形状）
P A H N
A P L S I I G
Y I R
之后按逐行顺序依次排列：”PAHNAPLSIIGYIR”

实现一个将字符串进行指定行数的转换的函数:
string convert(string text, int nRows);
convert(“PAYPALISHIRING”, 3) 应当返回 “PAHNAPLSIIGYIR” 。

解题思路

使用二维数组，按照Z字形的方式进行排序，最终，输出结果。

Java

public String convert(String s, int numRows) {
        StringBuffer[] stringBuffers = new StringBuffer[numRows];
        for (int i = 0; i < numRows; i++)
            stringBuffers[i] = new StringBuffer();
        int i = 0;
        while(i<s.length()){
            for(int j = 0;j<numRows&&i<s.length();j++)
                stringBuffers[j].append(s.charAt(i++));
            for(int j=numRows-2;j>=1&&i<s.length();j--)
                stringBuffers[j].append(s.charAt(i++));
        }
        for(i=1;i<numRows;i++)
            stringBuffers[0].append(stringBuffers[i]);
        return stringBuffers[0].toString();
    }

python

class Solution(object):
    def convert(self, s, numRows):
        N = [[] for i in range(numRows)]
        i = 0
        while (i < len(s)):
            for j in range(numRows):
                if i >= len(s):
                    break
                N[j].append(s[i])
                i = i + 1
            for j in range(numRows - 2, 0, -1):
                if i >= len(s):
                    break
                N[j].append(s[i])
                i = i + 1
        result = ""
        for i in range(numRows):
            for j in range(len(N[i])):
                result = result + N[i][j]
        return result

C++

class Solution
{
public:
    string convert(string s, int numRows)
    {
        vector<string>vec(numRows, "");
        int i=0;
        while(i<s.length())
        {
            for(int j=0; j<numRows&&i<s.length(); j++) vec[j].push_back(s[i++]);
            for(int j=numRows-2; j>=1&&i<s.length(); j--) vec[j].push_back(s[i++]);
        }
        string result;
        for(i=0; i<numRows; i++)
        {
            result+=vec[i];
        }
        return result;
    }
};