【算法day8】 Z 字形变换 （O2算法思路整理2）

Z 字形变换，算法思路整理

https://leetcode.cn/problems/zigzag-conversion/

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

比如输入字符串为 “PAYPALISHIRING” 行数为 3 时，排列如下：
P A H N
A P L S I I G
Y I R
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：“PAHNAPLSIIGYIR”。

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

• 不擅长打印图形，我将把所有矩阵和一维坐标的映射关系都学习一遍

class Solution {
public:
    string convert(string s, int numRows) {
        int s_len = s.size();
        if (numRows == 1 || numRows >= s_len) {
            return s;
        }
        int T = numRows + numRows - 2;
        int T_length = numRows - 1;
        int T_num = s_len / T + 1;
        int numCols = T_length * T_num;
        char* ch = (char*)calloc(numRows * numCols, sizeof(char));
        for (int i = 0, j = 0, k = 0; k < s_len; k++) {
            if (k % T < T_length) {
                // 向下移动
                ch[i * numCols + j] = s[k];
                i++;
            } else {
                // 向右上方移动
                ch[i * numCols + j] = s[k];
                j++;
                i--;
            }
        }
        string ans;
        for (int i = 0, j = 0; i < numCols * numRows; i++) {
            if (ch[i] && ch[i] != '\0') {
                ans+= ch[i];
                j++;
            }
        }
        return ans;
    }
};