Leetocde 6. Z 字形变换

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：
P   A   H   N
A P L S I I G
Y   I   R
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。
请你实现这个将字符串进行指定行数变换的函数：
string convert(string s, int numRows);

示例 1：
输入：s = "PAYPALISHIRING", numRows = 3
输出："PAHNAPLSIIGYIR"

示例 2：
输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"
解释：
P     I    N
A   L S  I G
Y A   H R
P     I

示例 3：
输入：s = "A", numRows = 1
输出："A"

提示：
1 <= s.length <= 1000
s 由英文字母（小写和大写）、',' 和 '.' 组成
1 <= numRows <= 1000

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>


using namespace std;

//class Solution {
//public:
//    string convert(string s, int numRows) {
//        if (numRows < 2) return s;
//        string str;
//        for (int j = 0; j < min(numRows , (int)s.size()); j++) {
//            int i = j;
//            while (i < s.size()) {
//                str += s[i];  int jump = numRows + numRows - 2;
//                if (j !=0 && j!= numRows-1 && i + jump - j * 2 <s.size()) {
//                    str += s[i+ jump -j*2];
//                }
//                i += jump;
//            }
//        }
//        
//        cout << str << endl;
//
//        return str;
//    }
//};


class Solution {
public:
    char mm[1010][1010];
    string convert(string s, int numRows) {
        if (numRows < 2) return s;
        memset(mm, 0, sizeof mm);
        int currx = 0; int curry = 0;
        int down = 1; int up = 0;
        for (int i = 0; i < s.size(); i++) {
            mm[currx][curry] = s[i]; 
            if (i % (numRows * 2 - 2) == 0) {
                down = 1; up = 0;
            }
            else if (i % (numRows-1) == 0) {
                up = 1; down = 0;
            }
            if (down == 1) {
                currx++;
            }
            else {
                currx--; curry++;
            }
        }
        string ans;
        for (int i = 0; i < numRows; i++) {
            for (int j = 0; j < 1010; j++) {
                if ( mm[i][j]!=0) {
                    ans += mm[i][j];
                    //cout << mm[i][j];
                }
            }
            //cout << endl;
        }


        return ans;
    }
};



int main()
{
    Solution s;
    s.convert("PAYPALISHIRING", 4);

    return 0;
}