[LeetCode] #6 Z 字形变换

将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时,排列如下:

 之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"PAHNAPLSIIGYIR"

请你实现这个将字符串进行指定行数变换的函数:

string convert(string s, int numRows);

输入:s = "PAYPALISHIRING", numRows = 4

输出:"PINALSIGYAHRPI"

解释:

使用List<StringBuilder>来模拟Z形变换后的每一行,依次递增存到每一行,第一个第一行、第二个第二行.......当row等于0或numRows-1时,转向变成依次递减,第四个第四行、第五个第三行.....

class Solution {
    public String convert(String s, int numRows) {
        if (numRows == 1) return s;
        List<StringBuilder> list = new ArrayList<>();
         for (int i = 0; i < Math.min(numRows, s.length()); i++)
            list.add(new StringBuilder());
        boolean flag = false;
        int row = 0;
        for(char ch: s.toCharArray()){
            list.get(row).append(ch);
            if(row == 0 || row == numRows-1) flag = !flag;
            if(flag) row++;
            else row--;
        }
        StringBuilder res = new StringBuilder();
        for (StringBuilder str: list) res.append(str);
        return res.toString();
    }
}

也可以找规律

class Solution {
    public String convert(String s, int numRows) {
        if (numRows == 1) return s;
        StringBuilder res = new StringBuilder();
        int cycleLen = 2 * numRows - 2;
        for (int i = 0; i < numRows; i++) {
            for (int j = 0; j + i < s.length(); j += cycleLen) {
                res.append(s.charAt(j + i));
                if (i != 0 && i != numRows - 1 && j + cycleLen - i < s.length())
                    res.append(s.charAt(j + cycleLen - i));
            }
        }
        return res.toString();
    }
}

知识点:

总结:

posted @ 2021-09-24 21:40  1243741754  阅读(39)  评论(0编辑  收藏  举报