(字符串)leetcode6. N 字形变换

一、题目

1、题目描述

将一个给定字符串 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”

2、基础框架

• C++版本给出的基础框架如下：

3、原题链接

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

二、解题报告

1、思路分析

  $(1)$找到每一行所有元素下标的规律
  $(2)$当numRows=3时，第一行下标为0,4,8,12…第二行下标为1,3,5,7…第三行下标为2,6,10,14…
  $(3)$当numRows=4时，第一行下标为0,6,12,18…第二行下标为1,5,7,11…第三行下标为2,4,8,10…第四行下标为3,9,15,21…
  $(4)$当numRows=5时，第一行下标为0,8,16,24…第二行下标为1,7,9,15…第三行下标为2,6,10,14…第四行下标为3,5,11,13,19…第五行下标为4,12,20,28…
  $(5)$可以发现，第一行和最后一行的元素下标间距为2*numRows-2，设为n。第i行(非第一行最后一行)的下标间距为n-2i,2i,n-2i,2i
  $(6)$根据这个规律，我们可以依次取出每一行的元素

2、时间复杂度

时间复杂度为O(n)，空间复杂度为O(1)

3、代码详解

class Solution {
public:
    string convert(string s, int numRows) {
        string re = "";
        int n = 2*numRows-2;
        if (numRows == 1) return s;
        for(int i=0;i<numRows;i++){
            int k = i;
            int cnt = 0;
            while(k < s.length()) {
                re += s[k];
                if (i == 0 || i == numRows-1){
                    k = k + n;
                }
                else if (cnt%2 == 0) {
                    k = k + n-2*i;
                } else if (cnt%2 == 1){
                    k = k + 2*i;
                }
                cnt++;
            }
        }
        return re;
    }
};

三、本题小知识