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