LeetCode：Z 字形變換
字串 Z 字形變換的巧妙解法：設立標志位控制迭代方向
No.6 Z 字形變換
題目：
將一個給定字串根據給定的行數，以從上往下、從左到右進行 Z 字形排列，
示例 1:
比如輸入字串為 "LEETCODEISHIRING" 行數為 3 時，排列如下：
L C I R
E T O E S I I G
E D H N
之后，你的輸出需要從左往右逐行讀取，產生出一個新的字串，比如："LCIRETOESIIGEDHN"，
示例 2:
輸入: s = "LEETCODEISHIRING", numRows = 3
輸出: "LCIRETOESIIGEDHN"
示例 3:
輸入: s = "LEETCODEISHIRING", numRows = 4
輸出: "LDREOEIIECIHNTSG"
L D R
E O E I I
E C I H N
T S G
請你實作這個將字串進行指定行數變換的函式：string convert(string s, int numRows)
解法：
-
標志位
每行用一個 StringBuilder 遍歷裝入，設立一個 flag 標志位控制裝入的迭代方向
時間復雜度：\(O(n)\)
空間復雜度：\(O(n)\)
class Solution { public String convert(String s, int numRows) { if (numRows < 2) return s; int len = s.length(); ArrayList<StringBuilder> list = new ArrayList<>(); for (int i = 0; i < numRows; i++) list.add(new StringBuilder()); int count = 0, flag = 1; for (char c : s.toCharArray()) { list.get(count).append(c); if (count == 0) flag = 1; else if (count == numRows - 1) flag = -1; count += flag; } StringBuilder sb = new StringBuilder(); for (StringBuilder str : list) sb.append(str); return sb.toString(); } }
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/75820.html
標籤：其他