【LeetCode with Python】 ZigZag Conversion

最新推荐文章于 2026-05-22 13:54:45 发布

原创最新推荐文章于 2026-05-22 13:54:45 发布 · 1.2w 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

LeetCode with Python

针对LeetCode上的一道关于字符串处理的问题，文章介绍了一种算法，该算法将输入的字符串以Z字形的方式进行排列，并按行读取返回。通过巧妙地计算下标并控制循环条件，确保了下标不会越界。

题目

原题页面：https://leetcode.com/problems/zigzag-conversion/
本文地址：http://www.xnerv.wang/leetcode/zigzag-conversion/
题目类型：String
难度评价：Easy

The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)
P---A---H---N
A-P-L-S-I-I-G
Y---I---R

And then read line by line: "PAHNAPLSIIGYIR"
Write the code that will take a string and make this conversion given a number of rows:
string convert(string text, int nRows);
convert("PAYPALISHIRING", 3) should return "PAHNAPLSIIGYIR".

分析

下标计算处理，控制好循环条件，下标不要越界即可。

代码

class Solution:
    # @return a string
    def convert(self, s, nRows):
        if 1 == nRows:
            return s

        len_s = len(s)
        piece_len = 2 * (nRows - 1)

        result = ""
        index = 0
        while index < len_s:
            result += s[index]
            index += piece_len
        for m in range(1, nRows - 1):
            index = m
            while index < len_s:
                result += s[index]
                right_index = index + (nRows - m - 1) * 2
                if right_index < len_s:
                    result += s[right_index]
                index += piece_len
        index = nRows - 1
        while index < len_s:
            result += s[index]
            index += piece_len

        return result