字符串Z字形排列转换算法-优快云博客

本文链接：https://blog.youkuaiyun.com/fffffall/article/details/131642324

题目

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

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：

P   A   H   N
A P L S I I G
Y   I   R

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

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

string convert(string s, int numRows);

示例 1：

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

示例 2：

输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"
解释：
P     I    N
A   L S  I G
Y A   H R
P     I

示例 3：

输入：s = "A", numRows = 1
输出："A"

解答（直接法）

class Solution {
public:
    string convert(string s, int numRows) {
        int n = s.length();
        if(numRows==1) {
            return s;
        }
        string ans;
        int t = numRows * 2 - 2;
        for (int i = 0;i<numRows;i++) {
            for(int j = 0;j+i < n;j+=t) {
                ans+=s[j+i];
                if(i>0&&i<numRows-1;j-i+t<n) {
                    ans+=s[j+t-i];
                }
            }
        }
        return ans;
    }
}

周期的计算方法：

P           A           H           N
A P         L S         I I         G
Y           I           R

示例1：每四个为一组4=2*3-2

P             I            N
A   L         S  I         G
Y A           H R
P             I

示例2：每六个为一组6=2*4-2

不难得出：t=2*numRows-2

for循环里的内容：

0               0+t               0+2t                  0+3t
1        0+t-1  1+t        t+t-1  1+2t         2t+t-1   1+3t
2  0+t-2        2+t  t+t-2        2+2t  2t+t-2          2+3t  
3               3+t               3+2t                  3+3t

将0，1，2，3替换为i即可，同理可得示例1