Leetcode- Z 字形变换(6)

题目描述
将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “PAYPALISHIRING” 行数为 3 时，排列如下：
P A H N
A P L S I I G
Y I R
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：“PAHNAPLSIIGYIR”。
请你实现这个将字符串进行指定行数变换的函数：
string convert(string s, int numRows);

输入：s = “PAYPALISHIRING”, numRows = 3
输出：“PAHNAPLSIIGYIR”

解题思路
1.如果这个行数是没有从超过三行，那么直接返回这个字符串就行了，即

if (numRows < 2) return s #从0 开始

2.其实我们就是按行来放，那我们也按行来放入对应的数组里就是了，即就是第一行的我们就放进
res[0],第二行放进res[1]…以此类推

但是我们要考虑Z字形，即到头了，要换行的，即当i = 0 或者 i = numRows -1 的时候要换行
我们可以用flag 来表示，即当需要换行，我们让flag = - flag

3.最后把字符串加在一起就是 最后得到的字符串了

算法流程

res[i] += c： 把每个字符 c 填入对应行  ；
i += flag： 更新当前字符 c 对应的行索引；
flag = - flag： 在达到 Z 字形转折点时，执行反向。

代码
Java

class Solution {
    public String convert(String s, int numRows) {
        if(numRows < 2) return s;
        List<StringBuilder> rows = new ArrayList<StringBuilder>();
        for(int i = 0;i<numRows;++i){
            rows.add(new StringBuilder());
        }
        int i = 0,flag = -1;
        for(char c : s.toCharArray()){
            rows.get(i).append(c);
            if(i == 0 || i == numRows -1) flag =-flag;
            i += flag;
        }
        StringBuilder res = new StringBuilder();
        for(StringBuilder row : rows) res.append(row);
        return res.toString();

    }
}

Python

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows < 2:
            return s
        i,flag = 0,-1
        res = ["" for _ in range(numRows)]
        for c in s:
            res[i] += c
            if i == 0 or i == numRows -1:
                flag = -flag
            i += flag
        return "".join(res)