LeetCode-6 Z字形變換
阿新 • • 發佈:2022-03-01
來源:力扣(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
執行結果