leetcode 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

思路：

我最先想到的就两种方法就是直接按照题目的描述，先将字符串存存储到二维数组中，然后再按行提取出来；二是找规律，但是我试了下老有问题所以我选择使用第一种笨方法。

//1.将字符串存储到一个行数为numRows的二维数组，浪费空间且麻烦
//2.直接找到规律
char arr[1000][1000]  = {0};
char * convert(char * s, int numRows){
    if(numRows == 1)//一行的时候直接输出
        return s;
    for(int i = 0; i < numRows; i++) //先清空需要的行数，方便下面提取字符时判断
        memset(arr+i, 0, 1000);
    int len =  strlen(s);
    int flag = 1, k = 0;    //flag用来标记是向下走，还是向斜上走
    int x = 0, y = 0;
    for(int i = 0; i < len; i++){
        if(flag){
            arr[x++][y] = s[i];
            if(x == numRows - 1)   //走到行数最大值时要向斜上走
                flag = 0;
        }else{
            arr[x--][y++] = s[i];    
            if(x == 0)        //行数最到0后要向下走
                flag = 1;
        }
    }
    char *ans = malloc(1020);
    for(int i = 0; i < numRows; i++){    //按行遍历
        for(int j = 0; j <= y; j++){
            if(arr[i][j])                //如果存储的有字符，则就是我们需要的字符
                ans[k++] = arr[i][j];
        }
    }
    ans[len ] = '\0';        //结尾加'\0'
    return ans;
}