【算法day8】 Z 字形变换 -O(n）算法思路整理

舔甜歌姬的EGUMI LEGACY2025-03-13 9:56

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"。

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

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

cpp 复制代码

class Solution {
public:
    string convert(string s, int numRows) {
        int s_len = s.size();
        if (numRows >= s_len || numRows == 1)
            return s;
        int T_length = numRows + numRows - 2;
        string ans;
        for (int i = 0; i < numRows; i++) {
            for (int j = 0; i + j < s_len; j = j + T_length) {
                ans += s[i + j];
                // 不是首行和最后一行的时候，
                if (i > 0 && i < numRows - 1 && j + T_length - i < s_len) {
                    ans += s[j + T_length - i];
                }
            }
        }
        return ans;
    }
};