C++&Python 描述 LeetCode 6. Z 字形变换

Posted 2021-06-27 亓官劼

C++&Python 描述 LeetCode 6. Z 字形变换

  大家好，我是亓官劼（qí guān jié ），在【亓官劼】公众号、CSDN、GitHub、B站、华为开发者论坛等平台分享一些技术博文。放弃不难，但坚持一定很酷！时光荏苒，未来可期，加油~

  如果喜欢博主的文章可以关注博主的个人公众号【亓官劼】(qí guān jié)，如果有需要找博主的话可以在公众号后台留言。建了个小交流群，Q群：545611263，要进微信群的话可以加我V:qiguanjie2015备注拉群即可。

  《leetcode刷题系列》，博主将从第一题开始，每天更新当日的刷题情况，每题博主会尽量加一些解题思路的分析和C++&python代码的实现。每日在微信公众号【亓官劼】定时更新，有兴趣一起刷题的小伙伴可以关注公众号来一起卷起来呀。

  同时文章在GitHub中进行了开源，内含本系列文章目前已刷的各个题解和解题思路，GitHub地址为：LeetCode，如果文章对你有帮助的话可以来GitHub点个star，如果有更好的解题思路的话，也可以来GitHub提交~一起改进

题目

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

提示：

• 1 <= s.length <= 1000
• s 由英文字母（小写和大写）、',' 和 '.' 组成
• 1 <= numRows <= 1000

解题思路

模拟。我们建立numRows个 string 类型的数组，从字符串第一个位置开始遍历，到头后转弯，依次将字符加入到各行，最后汇总即可。

算法实现 C++

class Solution {
public:
    string convert(string s, int numRows) {
        vector<string> tmp(numRows+1);
        bool flag = false;
        int len = s.length(), idx = 0;
        for(int i = 0; i < len ; i++){
            tmp[idx] += s[i];
            if(idx == 0 || idx == numRows-1) flag = !flag;
            idx = flag ? (idx + 1) % numRows  : (idx - 1) % numRows;
        }
        string res = "";
        for(int i = 0; i < numRows; i++) res += tmp[i];
        return res;
    }
};

算法实现 Python

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        tmp = ['' for i in range(s.__len__())]
        flag = False
        length = s.__len__()
        idx = 0
        for i in range(length):
            tmp[idx] += s[i]
            if idx == 0 or idx == numRows - 1: flag = ~flag
            idx = (idx + 1) % numRows if flag else (idx - 1) % numRows
        res = ''.join(tmp)
        return res