Leetcode刷题-ZigZag

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 s, int numRows);

官方的解决方案

class Solution {
public:
    string convert(string s, int numRows) {

        if (numRows == 1) return s;

        string ret;
        int n = s.size();
        int cycleLen = 2 * numRows - 2;

        for (int i = 0; i < numRows; i++) {
            for (int j = 0; j + i < n; j += cycleLen) {
                ret += s[j + i];
                if (i != 0 && i != numRows - 1 && j + cycleLen - i < n)
                    ret += s[j + cycleLen - i];
            }
        }
        return ret;
    }
};

// 我的解决方案
// Runtime: 56 ms, faster than 27.27% of C++ online submissions for ZigZag Conversion.
// Memory Usage: 14.1 MB, less than 18.64% of C++ online submissions for ZigZag Conversion.
class Solution {
public:
    string convert(string s, int numRows) {
        if(s.length() == 0 or numRows <= 1)
            return s;
        int row = numRows;
        int col = numRows-1;
        int flag[row][col];
        memset( flag, 0, sizeof(flag));
        for(int i = 0;i < row;i++){
            flag[i][0] = 1;
        }
        for(int i = 1; i < col;i++){
            int j = row-i-1;
            flag[j][i] = 1;
        }
  
        vector<string> vstr(numRows,"");
        int cur = 0;
        int m = 0;
        int n = 0;
        int loop = row*col;
        int loopCnt = 0;
        while(cur < s.length()){
            if(flag[m][n] == 1){
                vstr[m] += s[cur++];
            }
            m++;
            if(m >= row){
                m = 0;
                n++;
            }
            loopCnt++;
            if(loopCnt >= loop){
                m = 0;
                n = 0;
                loopCnt = 0;
            }
        }

        string rs = "";
        for(int i = 0;i < numRows;i++){
            rs += vstr[i];
        }
  
        return rs;
    }
};
// 总结：我的思路还是不清晰，方法也不够高级，做题太少了的缘故吧