Z 字形变换

LeetCode 6.Z 字形变换

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

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

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

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

  请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

  示例 1:

输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"

  示例 2:

输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"

  解释:

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

  打卡：

char * convert(char * s, int numRows){
    int len = strlen(s);
    char *arr = (char *)malloc(sizeof(char) * (len + 1));
    if(numRows == 1)
        strncpy(arr, s, len);
    else if(numRows == 2) {
        int j = 0;
        for(int i = 0; i < len; i += 2)
            arr[j++] = s[i];
        for(int i = 1; i < len; i += 2)
            arr[j++] = s[i];
    }
    else {
        int j = 0, row = 2 * numRows - 2;
        for(int i = 0; i < len; i += row)
            arr[j++] = s[i];
        for(int i = 1; i < numRows - 1; ++i) {
            int k1 = row - i * 2;
            int k2 = 2 * i;
            for(int h = i; h < len; ) {
                arr[j++] = s[h];
                h += k1;
                if(h < len) {
                    arr[j++] = s[h];
                    h += k2;
                }
            }
        }
        for(int i = numRows - 1; i < len; i += row)
            arr[j++] = s[i];
    }
    arr[len] = '\0';
    return arr;
}