leetcode-6.Z 字形变换

r囧r小猫 2021-11-17 06:18 269阅读 0赞

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

# 题目 #

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

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

L C I R  
E T O E S I I G  
E D H N  
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：“LCIRETOESIIGEDHN”。

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

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

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

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

L D R  
E O E I I  
E C I H N  
T S G

class Solution { 
    public:
        string convert(string s, int numRows) { 
    
            if (numRows == 1) return s;
    
            vector<string> rows(min(numRows, int(s.size())));
            int curRow = 0;
            bool goingDown = false;
    
            for (char c : s) { 
                rows[curRow] += c;
                if (curRow == 0 || curRow == numRows - 1) goingDown = !goingDown;
                curRow += goingDown ? 1 : -1;
            }
    
            string ret;
            for (string row : rows) ret += row;
            return ret;
        }
    };

利用goingDown变量，初始化为false。如果当前行为第一行（curRow==0）或是最后一行（curRow == numRows - 1），则令goingDown取反。三目运算符根据goingDown是true还是false，判别curRow是加1还是减1。  
C++最妙的就是不像C一样需要精确到数组下标，直接+就可以。  
菜鸟记录一下叹为观止的官方题解。

[https_leetcode-cn.com_problems_zigzag-conversion]: https://leetcode-cn.com/problems/zigzag-conversion