6. Z 字形变换

题目描述

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。 比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：

P    A    H    N
A P L S  I  I  G
Y     I     R
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。

示例

示例1：
输入：s = "PAYPALISHIRING", numRows = 3
输出："PAHNAPLSIIGYIR"

示例2：
输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"

来源：力扣（LeetCode）
链接：leetcode-cn.com/problems/lo…

实现

规律：

1. numRows(后面简写n)代表行数，也是列的最大元素个数，n=3，那么第0列的元素就为3个元素
2. 根据示例可以看出，假设有n行，第0列有n个元素，后面跟n-2个包含1个元素的列。将这n-1个列一起认为是一个组合列，可知1个组合列中包含的元素为n+(n-2)=2n-2个
3. 此时可以对字符串进行处理，遍历字符串，对字符串的元素分为2中情况

• 第一种情况如第0列，从上往下遍历
• 第二种情况是后面跟的n-2个包含1个元素的列，从下往上遍历 举例：
  以字符串"PAHNAPLSIIGYIR"为例 第一种情况是PAY,ALI,HIR,NG这种情况，每个元素对2n-2求余，余数就是行数 第二种情况是PSI这种情况，从下往下走，行数与目前下标相反，每个元素对2n-2求余，此时行数2n-2-余数就是行数。

char *convert(char * s, int numRows)
{
    if (s == NULL) {
        return NULL;
    }
    if (strlen(s) == 1 || numRows <= 1) {
        return s;
    }
    int idx = 0;
    char *res = (char*)malloc(sizeof(char) * (strlen(s) + 1));
    int n = 2 * numRows - 2;
    for (int i = 0; i < numRows; i++) {
        for (int j = 0; j < strlen(s); j++) {
            if (j % n == i || j % n == n - i) {
                res[idx++] = s[j];
            }
        }
    }
    res[idx] = '\0';
    return res;
}