Z字形变换

题目描述

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时，排列如下：

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：“LCIRETOESIIGEDHN”。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);
示例 1:

输入: s = “LEETCODEISHIRING”, numRows = 3
输出: “LCIRETOESIIGEDHN”
示例 2:

输入: s = “LEETCODEISHIRING”, numRows = 4
输出: “LDREOEIIECIHNTSG”
解释:

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zigzag-conversion
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路1

根据变换后Z字型的行数构建一个二维字符数组，数组每一维对应Z字型每行的字符串。顺序遍历字符串s，根据方向变量orient指示的方向将遍历到的字符添加到对应行的字符串中。当在第0行时，orient指示向下顺序插入二维字符数组；当在第numRows-1行时，orient指示向上顺序插入二维字符数组。

代码（c++）

class Solution {
public:
    string convert(string s, int numRows) {
        string res;
        if(s.length()==0||numRows==0) return res;
        if(numRows==1) return s;
        vector<string> rows(min(int(s.length()),numRows));
        bool orient=false;
        int r=0;
        for(int i=0;i<s.length();i++){
            rows[r]+=s[i];
            if(r==0||r==numRows-1) orient=!orient;
            r+=orient?1:-1;
        }
        for(int i=0;i<rows.size();i++){
            res+=rows[i];
        }
        return res;
    }
};

思路2

寻找变换后Z字形第一行的字符对应原字符串s下标的关系。通过观察可以发现，在第一行中每个字符之间间隔2*numRows-3个字符。于是我们可以根据Z字形第一行的字符在原字符串中的位置来找到剩下Z字形中的字符在原字符串中的位置。
1.先找出第一行的所有字符，s[0],s[2*numRows-2],…,s[k*(2*numRows-2)]；

2.对于第二行，和第一行在Z字形中同一列的字符对应原字符串中第一行的字符的下一个字符，于是对于和第一行同列的字符有s[0+1],s[2*numRows-2+1],…,s[k*(2*numRows-2)+1]；在列与列之间，还有一个字符，是Z字形中第一行字符在原字符串中的前一个字符，于是有s[2*numRows-2-1],s[2*(2*numRows-2)-1],…,s[k*(2*numRows-2)-1]；所以最终Z字形第二行的字符对应原字符串的字符为s[0+1],s[2*numRows-2-1],s[2*numRows-2+1],s[2*(2*numRows-2)-1],s[2*(2*numRows-2)+1]…,s[k*(2*numRows-2)-1],s[k*(2*numRows-2)+1]

3.根据2.，对于第i行(i<numRows)，有s[0+i],s[2*numRows-2-i],s[2*numRows-2+i],s[2*(2*numRows-2)-i],s[2*(2*numRows-2)+i]…,s[k*(2*numRows-2)-i],s[k*(2*numRows-2)+i]

4.对于最后一行，因为最后一行列与列之间没有元素，所以有s[0+numRows],s[2*numRows-2+numRows],…,s[k*(2*numRows-2)+numRows]；

代码（c++）

class Solution {
public:
    string convert(string s, int numRows) {
        string res;
        if(s.length()==0||numRows==0) return res;
        if(numRows==1) return s;
        int stride=2*numRows-2;
        for(int i=0;i<numRows;i++){
            for(int j=0;j+i<s.length();j+=stride){
                res+=s[j+i];
                if(i!=0&&i!=numRows-1&&j+stride-i<s.length()){
                    res+=s[j+stride-i];
                }
            }
        }
        return res;
    }
};