1. 程式人生 > 其它 >LeetCode-6 Z字形變換

LeetCode-6 Z字形變換

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/zigzag-conversion

題目描述

將一個給定字串 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

解題思路

第一次每日一題刷到當年做過的題目。

用二維連結串列進行簡單的模擬,根據題目規則將原字串z字排列寫入二維連結串列中,然後依次拼接二維連結串列。

程式碼展示

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        #自己的做法,效率感人

        zhe = numRows - 1
        news 
= [] if numRows < 2: return s for i in range(numRows): news.append("") b = True p = 0 k = 0 while p < len(s): news[k] += s[p] p += 1 if k == 0 or k == zhe: b =not b if b: k
-= 1 else: k += 1 re = "".join(news) return re

執行結果