LeetCode - Z 字形变换

题目链接：https://leetcode-cn.com/problems/zigzag-conversion
博客：http://www.coolsite.top/archives/28

题目描述

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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

我的思路

• 做完看了下其他人的答案，感觉大部分人都把这道题搞得有点复杂了。
• 这道题可以这么考虑：对于输入的字符串s，其下标 i / (numRows-1) 如果为偶数或零，则代表该下标代表的字符在Z字型中属于竖列。如果 i / (numRows-1) 如果为奇数，则代表该下标代表的字符在Z字型中属于斜列（Z字型的中间倾斜部分）。
• 如果当前字符属于竖列，则按照正序依次保存在字符串数组 temp 中（temp[remain].push_back(s[i])），如果属于斜列，则逆序保存（temp[numRows-remain-1].push_back(s[i])）。
• 最后将 temp 数组依次按序输出就是最终答案。

class Solution {
public:
    string convert(string s, int numRows) {
        vector<string> temp(numRows);
        string res;
        if(s.empty() || numRows < 1) return res;
        if(numRows == 1) return s;
        for(int i = 0; i < s.size(); i++){
            int ans = i / (numRows-1);
            int remain = i % (numRows-1);
            if(ans % 2 == 0){                           //结果为偶数或0
                temp[remain].push_back(s[i]);              //按余数正序保存
            }
            if(ans % 2 != 0){                           //结果为奇数
                temp[numRows-remain-1].push_back(s[i]);    //按余数倒序保存
            }
        }
       for(int i = 0; i < temp.size(); i++){
               res += temp[i];
        }
        return res;
    }
};