6. Z 字形变换
將一個給定字符串根據給定的行數,以從上往下、從左到右進行?Z 字形排列。
比如輸入字符串為 "LEETCODEISHIRING"?行數為 3 時,排列如下:
L C I R E T O E S I I G E D H N之后,你的輸出需要從左往右逐行讀取,產生出一個新的字符串,比如:"LCIRETOESIIGEDHN"。
請你實現這個將字符串進行指定行數變換的函數:
string convert(string s, int numRows);示例?1:
輸入: s = "LEETCODEISHIRING", numRows = 3 輸出: "LCIRETOESIIGEDHN"示例?2:
輸入: s = "LEETCODEISHIRING", numRows =?4 輸出:?"LDREOEIIECIHNTSG" 解釋:L D R E O E I I E C I H N T S G解題思路:
- 題目理解:
- 字符串 s 是以 ZZZ 字形為順序存儲的字符串,目標是按行打印。
- 設 numRows 行字符串分別為 s1s_1s1? , s2s_2s2? ,..., sns_nsn?,則容易發現:按順序遍歷字符串 s 時,每個字符 c 在 ZZZ 字形中對應的 行索引 先從 s1s_1s1? 增大至 sns_nsn?,再從 sns_nsn? 減小至 s1s_1s1? …… 如此反復。
- 因此,解決方案為:模擬這個行索引的變化,在遍歷 s 中把每個字符填到正確的行 res[i] 。
- 算法流程: 按順序遍歷字符串 s;
- res[i] += c: 把每個字符 c 填入對應行 sis_isi?;
- i += flag: 更新當前字符 c 對應的行索引;
- flag = - flag: 在達到 ZZZ 字形轉折點時,執行反向。
- 復雜度分析:
- 時間復雜度 O(N)O(N)O(N) :遍歷一遍字符串 s;
- 空間復雜度 O(N)O(N)O(N) :各行字符串共占用 O(N)O(N)O(N) 額外空間。
代碼:
class Solution:def convert(self, s: str, numRows: int) -> str:if numRows < 2: return sres = ["" for _ in range(numRows)]i, flag = 0, -1for c in s:res[i] += cif i == 0 or i == numRows - 1: flag = -flagi += flagreturn "".join(res) class Solution {public String convert(String s, int numRows) {if(numRows < 2) return s;List<StringBuilder> rows = new ArrayList<StringBuilder>();for(int i = 0; i < numRows; i++) rows.add(new StringBuilder());int i = 0, flag = -1;for(char c : s.toCharArray()) {rows.get(i).append(c);if(i == 0 || i == numRows -1) flag = - flag;i += flag;}StringBuilder res = new StringBuilder();for(StringBuilder row : rows) res.append(row);return res.toString();} }總結
- 上一篇: LeetCode Algorithm 3
- 下一篇: LeetCode Algorithm 7