Z字形变换

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

 我们可维护numRows个字符数组来存储每行的字符,按照规则,如果numRows为1直接返回s,否则就自上而下,从下往上周期性对每个字符数组添加字符。

class Solution {
public:
    string convert(string s, int numRows) {
        vector<vector<char>>p;
        vector<char> a;
        for(int i=0;i<numRows;i++)
            p.push_back(a);
        int count=0,i=0,flag=1;
        if(numRows==1)return s;
        while(count<s.size()){
            if(1){
                p[i].push_back(s[count++]);
                if(flag)i++;
                else i--;
                if(i==numRows){
                    flag=0;
                    i-=2;
                }
                if(i==-1){
                    flag=1;
                    i+=2;
                }
                
            }
        }
        string sl;
        for(int i=0;i<numRows;i++)
            for(int j=0;j<p[i].size();j++)
                sl.append(1,p[i][j]);
        return sl;
    }
};

上一篇:用数学规律找字母编排,去解决leedcode上的第6题


下一篇:杨辉三角?这才是真正的杨辉三角(leetcode算法0基础 第四例组合数)