leetcode-06-zigzag conversion-python

convert(“PAYPALISHIRING”, 3) should return “PAHNAPLSIIGYIR”.

将字符串以z型重新排列输出。OJ通过。

    def convert(self, s, numRows):
        """
        :type s: str
        :type numRows: int
        :rtype: str
        """
        a=['']*numRows
        if len(s)==1 or numRows==1:
            return s
        else:
            for i in range(1,len(s)+1):
                k=2*numRows-2
                m=i%k
                if 0<m<=numRows:
                    a[m-1]+=(s[i-1])
                if m>numRows and m <=k-1:
                    n=m-numRows
                    a[numRows-1-n]+=(s[i-1])
                if m==0:
                    a[1]+=(s[i-1])
        f=''.join(a)     
        return f

思路是找规律遍历字符串，按照索引值来计算在哪一行，对应那行的保存字符串，最后将所有行想加。
注意：

a=[1]*2
b=[[1]*2]
c=[[1]]*2
print(a,b,c)
print(len(a))
print(len(b))

[1, 1] [[1, 1]] [[1], [1]]
2
1
2
Press any key to continue . . .

如果想创建多维列表，用c的方式。
下面贴一个20ms过的cpp程序。

class Solution {
public:
    string convert(string s, int nRows) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function    
        if(nRows <= 1) return s;
        string ret;
        int zigsize = 2 * nRows - 2;
        for(int i = 0; i < nRows; ++i) {
            for(int base = i; ;base += zigsize) {
                if(base >= s.size())
                    break;
                ret.append(1,s[base]);
                if(i > 0 && i < nRows - 1) {
                    //middle need add ziggggging char
                    int ti = base + zigsize - 2 * i;
                    if(ti < s.size())
                        ret.append(1,s[ti]);
                }
            }
        }
        return ret;
    }
};

思路是一行行加s中特定元素。看起来两次循环。cpp就是比python快。