leetcode-Z 字形变换（c++）

题目

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

大体思路为：可以按顺序遍历字符串"LEETCODEISHIRING"，将字符填到相应位置的a[i]上。分别按序取出a[i]中的字符拼接，
“LDR”+“EOEII”+“ECIHN”+“TSG”=“LDREOEIIECIHNTSG”

• 可以用flag标记，当达到Z字转折点时flag=-flag
  （看了题解学会的，实在是妙啊）

• 可以考虑到若numRows=1或nunRows大于字符串s的长度则字符串s不变。

代码

class Solution {
public:
    string convert(string s, int numRows) {
        if(numRows==1||numRows>s.size()){
            return s;
        }
        vector<string> a(numRows);
        string res;
        int flag = -1;
        int tmp=0;
        for(int i =0 ;i<s.size();i++){
            a[tmp].push_back(s[i]);
            if(tmp==0||tmp==numRows-1){
                flag=-flag;
            }
            tmp+=flag;
        }
        for(int i=0;i<a.size();i++){
            res+=a[i];
        }
        return res;
    }
};