【算法day6】Z 字形变换

【算法day6】 Z 字形变换

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

将一个给定字符串 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 sep = numRows + numRows - 2;     // 字母间隔
        int firstLine = (s_len + sep) / sep; // 第一行长度
        int remain = s_len % sep;
        int numCols = (firstLine - 1) * (1 + numRows - 2) + remain; // 矩阵宽度

        vector<char> ch(numCols * numRows);
        // char* ch = (char*)calloc(numCols * numRows, sizeof(char));
        int up_sep = numRows - 1;
        int index_s = 0;
        for (int j = 0; j < numCols; j++) {
            int remain_up_sep = up_sep;
            for (int i = 0; i < numRows; i++) {
                if (index_s > s_len) {
                    break;
                }
                if (up_sep != numRows - 1 && remain_up_sep != 0) {
                    remain_up_sep--;
                } else {
                    ch[i * numCols + j] = s[index_s];
                    index_s++;
                    if (remain_up_sep == 0) {
                        remain_up_sep--;
                    }
                }
            }
            up_sep--;
            if (up_sep == 0) {
                up_sep = numRows - 1;
            }
        }
        string ans;
        for (int i = 0; i < numRows; i++) {
            for (int j = 0; j < numCols; j++) {
                if (ch[i * numCols + j]) {
                    ans += ch[i * numCols + j];
                }
            }
        }
        return ans;
    }
};