Leetcode 6. Zigzag transformation (awesome, solved)

Will a given string s According to the given number of rows numRows , From top to bottom 、 Left to right Z Font arrangement .

For example, the input string is “PAYPALISHIRING” The number of rows is 3 when , Arranged as follows ：

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

after , Your output needs to be read line by line from left to right , Generate a new string , such as ：“PAHNAPLSIIGYIR”.

Please implement this function to transform a string into a specified number of lines ：

string convert(string s, int numRows);

Example 1：

 Input ：s = "PAYPALISHIRING", numRows = 3
 Output ："PAHNAPLSIIGYIR"

Example 2：

 Input ：s = "PAYPALISHIRING", numRows = 4
 Output ："PINALSIGYAHRPI"
 explain ：
P     I    N
A   L S  I G
Y A   H R
P     I

Example 3：

 Input ：s = "A", numRows = 1
 Output ："A"

Tips ：

• 1 <= s.length <= 1000
• s By the English letters （ Lowercase and upper case ）、‘,’ and ‘.’ form
• 1 <= numRows <= 1000

Code:

class Solution {
    
public:
    string convert(string s, int numRows) {
    
        if(numRows==1)
            return s;
        string res;
        int col=0;
        int loop=0;
        int row=numRows;
        vector<vector<char>> newOne(numRows, vector<char>(10000, ' '));
        do
        {
    
            //if(col)
            if(row==numRows)
            {
    
                for(int i=0;i<row;i++)
                {
    
                    newOne[i][col]=s[loop++];
                    if(loop==s.length())
                    {
    
                        loop=0;
                        for(int m=0;m<newOne.size();m++)
                        {
    
                            vector<char>sub=newOne[m];
                            for(int n=0;n<sub.size();n++)
                            {
    
                                if(sub[n]!=' ')
                                {
    
                                    loop++;
                                    res+=sub[n];
                                }
                                // cout<<sub[n]<<" ";
                                if(loop==s.length())
                                    return res;
                            }
                            // cout<<endl;
                        }
                        return res;
                    }
                }
            }
            else
            {
    
                newOne[row-1][col]=s[loop++];
                if(loop==s.length())
                {
    
                    
                    loop=0;
                    for(int m=0;m<newOne.size();m++)
                    {
    
                        vector<char>sub=newOne[m];
                        for(int n=0;n<sub.size();n++)
                        {
    
                            if(sub[n]!=' ')
                            {
    
                                loop++;
                                res+=sub[n];
                            }
                            // cout<<sub[n]<<" ";
                            if(loop==s.length())
                                return res;
                        }
                        // cout<<endl;
                    }
                    return res;
                }
            }
            col++;
            if(row>1)
                row--;
            else
                row++;
            if(row==1)
            {
    
                row=numRows;
            }
        }while(1);
        
        return res;
    }
};