LC6. 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

解题思路：

  本题考查的是找规律，很多题解都是找它的周期2n-2，我就比较简单粗暴了，直接把每一列的字符存进对应容器中，列最长肯定是numRows，中间一个一个的就从后往前存就行了，基本固定在numRows-2。

代码如下：

#include<iostream>
#include<vector> 
using namespace std;
string convert(string s, int numRows) {
        string a[numRows];
        if(s.size()==numRows){//字符串的大小等于numRows直接返回s
            return s;
        }
        for(int i=0;i<s.size();i++){
            for(int j=0;j<numRows&&i<s.size();j++,i++){//读取完整的一列，i记得要判断会不会出界
                a[j]+=s[i];
            }
            for(int j=numRows-2;j>=1&&i<s.size();j--,i++){//从后往前读取单独一个的一列
                a[j]+=s[i];
            }
            i--;//一次循环结束了会i++，上面结束循环已经+1了，所以要-1
        }
        string tmp="";
        for(int i=0;i<numRows;i++){//把保存的字符串都放到tmp中
            tmp+=a[i];
        }
        return tmp;
    }
int main(){
	string a="LEETCODEISHIRING";
	cout<<convert(a,4);
	return 0;
}