栈学习四：Z 字形变换

栈学习四：Z 字形变换

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);

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

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

解题思路（参考题解）

最开始题目都没看懂

题解解析

2.每一行字母的所有下标其实是有规则的
我们先假定有 numRows=4 行来推导下，其中 2numRows-2 = 6 , 我们可以假定为 step=2numRows-2 ，我们先来推导下规则：

第0行： 0 - 6 - 12 - 18

==> 下标间距 6 - 6 - 6 ==> 下标间距 step - step - step

第1行： 1 - 5 - 7 - 11 - 13

==> 下标间距 4 - 2 - 4 - 2 ==> 下标间距step-21(行)-21(行)-step-21(行)-21(行)

第2行： 2 - 4 - 8 - 10 - 14
==> 下标间距 2 - 4 - 2 - 4 ==> 下标间距step-22(行)-22(行)-step-22(行)-22(行)

第3行：3 - 9 - 15 - 21

==> 下标间距间距 6 - 6 - 6 ==>下标间距step - step - step

可以得出以下结论：

起始下标都是行号

第0层和第numRows-1层的下标间距总是step 。

中间层的下标间距总是step-2行数，2行数交替。

下标不能超过len(s)-1

自己敲的

//一行一行的判断，压入字符串
string convert(string s,int numRows)
{
    string str;
    int step;
    int a=0,h=0;//h:行号,a:每个元素需要加的长度
    step=2*numRows-2;
    //只有一行数据时
    if(step==0) return s;
    //用行号判断执行情况
    while(h<=numRows-1)
    {
        //第一行和最后一行的位置情况
        if(h==0||(h==(numRows-1)))
        {
            str.push_back(s[a]);
            a+=step;
        }
        else
        {
            //中间行的位置情况，有两种情况，加上step-2*行数或者加上2*行数
            str.push_back(s[a]);
            a+=step-2*h;
            if(a<s.length()) str.push_back(s[a]);
            a+=2*h;
        }
        //通过元素在字符串的位置判断是否换行
        //换行后，元素的位置为行号
        if(a>(s.length()-1))
        {
            h++;
            a=h;
        }

    }
    return str;
}