6.Z字形变换

6. Z 字形变换

将一个给定字符串 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);

示例 1：

输入：s = "PAYPALISHIRING", numRows = 3
输出："PAHNAPLSIIGYIR"

示例 2：

输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"
解释：
P     I    N
A   L S  I G
Y A   H R
P     I

示例 3：

输入：s = "A", numRows = 1
输出："A"

提示：

• 1 <= s.length <= 1000
• s 由英文字母（小写和大写）、',' 和 '.' 组成
• 1 <= numRows <= 1000

思路：

直接用模拟的方法。使用StringBuilder数组即可，大小为numRows,即每行都用一个StringBuilder表示，先竖着排，即分别给每个StringBuilder都存一个数，然后，再斜着往回排，即排除最后一个，StringBuilder和第一个StringBuilder，按逆序给其他StringBuilder存入一个数，如此循环，直到s中所有的字符都已经存完为止。

Java代码

class Solution {
    public String convert(String s, int numRows) {
        if(numRows <= 1) return s;
        StringBuilder[] sb = new StringBuilder[numRows];
        //注意刚刚只是初始化了数组sb，里面的StringBuilder元素并没有初始化，都是null
        //故需要对他们初始化，否则将出现空指针异常
        for(int i = 0;i<sb.length;i++){
            sb[i] = new StringBuilder();
        }
        int index = 0;
        //当s中还有字符时就循环
        while(index < s.length()){
            //排竖着的,注意还有一个index<s.length()条件，即s中字符排完就结束了
            for(int i = 0;i < numRows && index < s.length();i++){
                sb[i].append(s.charAt(index++));
            }
            //排斜着的，第一和最后一行不排，且从下往上排
            for(int i = numRows-2;i >= 1 && index < s.length();i--){
                sb[i].append(s.charAt(index++));
            }
        }
        //全部排完后，都存到sb[0]中，然后转为String
        for(int i = 1;i<numRows;i++){
            sb[0].append(sb[i]);
        }
        return sb[0].toString();
  }
}

Go代码

func convert(s string, numRows int) string {
    //使用Builder数组即可，大小为numRows,即每行都用一个Builder表示
	//先竖着排，即分别给每个Builder都存一个数，然后，再斜着往回排，
	//排除最后一个Builder和第一个Builder，按逆序给其他Builder存入一个数，
	//如此循环， 直到s中所有的字符都已经存完
    sbs := make([]strings.Builder,numRows)
    i := 0
    for i < len(s) {
        // 竖着排 , 注意还有一个i<len(s)条件，即s中字符排完就结束了
        for j := 0; j < numRows && i < len(s);j++ {
            sbs[j].WriteByte(s[i])
            i++
        }

        // 从下往上斜着排，需要排除最后一行和第一行, 注意还有一个i<len(s)条件，即s中字符排完就结束了
        for j := numRows - 2;j >= 1 && i < len(s);j--{
            sbs[j].WriteByte(s[i])
            i++
        }
    }

    // 排完后，将各builder中的合到一个builder中,然后转为字符串返回
    var res strings.Builder
    for j := 0;j < numRows;j++ {
        res.WriteString(sbs[j].String())
    }
    return res.String()
}