Leetcode第六题--Z 字形变换

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，排列如下：

L   C   I   R
E T O E S I I G
E   D   H   N


之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："LCIRETOESIIGEDHN"。

char * convert(char * s, int numRows){
    int len = strlen(s);
    if (len <= 1 || numRows <= 1)
        return s;
        
    char matrix[numRows][len];      //用来存储字符行列的矩阵
    memset(matrix, 0, sizeof(matrix));

    int num = 0;
    int column = 0;
    while (num < len)
    {
        if (column % (numRows - 1) == 0)
        {
            for (int j = 0; j < numRows && num < len; j++)
            {
                matrix[j][column] = s[num++];     //按照规律将字符串放入矩阵
            }
            column++;
        }
        else
        {
            matrix[numRows - 1 - column%(numRows-1)][column] = s[num++];
            column++;
        }
    }
    num = 0;
    //从矩阵中按照要求取出字符
    for (int i = 0; i < numRows; i++)
    {
        for (int j = 0; j < column; j++)
        {
            if (matrix[i][j] != 0)
            {
                s[num++] = matrix[i][j];
            }
        }
    }
    return s;
}