LeetCode 6:ZigZag Conversion

The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

P   A   H   N
A P L S I I G
Y   I   R

And then read line by line:  "PAHNAPLSIIGYIR"

Write the code that will take a string and make this conversion given a number of rows:

string convert(string text, int nRows);

convert("PAYPALISHIRING", 3)  should return  "PAHNAPLSIIGYIR" .

字符串 "PAYPALISHIRING" 在一个指定的数字下写成"之字形"的图案如下：（从上往下读，然后往右上读。。。以此类推，顺序是↓↗↓↗↓↗，注意数字是3）

P   A   H   N
A P L S I I G
Y   I   R

之后按行输出："PAHNAPLSIIGYIR" （每行从左往右读）

编写代码实现指定数字的"之字形"变换

string convert(string text, int nRows);

convert("PAYPALISHIRING", 3) 应当返回  "PAHNAPLSIIGYIR"

题目整体不难，建立numRows个字符串，判断每个字符应当放在哪个字符串中，最后拼起来就好

class Solution {
public:
    string convert(string s, int numRows) {
        string temp[numRows];
        if(numRows==1||numRows==0||numRows>=s.length()) return s;
        for(int i=0;i<s.length();i++)
        {
            int num=i%(2*numRows-2);
            if(num<=numRows-1) temp[num].push_back(s[i]);
            else temp[2*numRows-2-num].push_back(s[i]);
        }
        string ans;
        for(int i=0;i<numRows;i++)
            ans+=temp[i];
        return ans;
    }
};