每日算法-Z字形变换

题目

将一个给定字符串 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

解法

思路： 我们可以使用 numsRows 个列表来表示Z 字形图案中的非空行。从左到右迭代 s，将每个字符添加到合适的行。可以使用当前行和当前方向这两个变量对合适的行进行跟踪。只有当我们向上移动到最上面的行或向下移动到最下面的行时，当前方向才会发生改变。

class Solution {
    // 按行排序
    public String convert(String s, int numRows) {
        if(numRows == 1 || s.length() < numRows){
            return s;
        }
        List<StringBuilder> list = new ArrayList<>();
        for(int i = 0 ; i < numRows ; i++){
            list.add(new StringBuilder());
        }

        int currRow = 0;
        // 表示改变方向
        boolean change = false;
        for(char c : s.toCharArray()){
            list.get(currRow).append(c);

            // 只有第一行或者最后一行，才改变方向
            if(currRow == 0 || currRow == numRows-1){
                change = !change;
            }
            currRow += change ? 1 : -1;
        }
        
        StringBuilder result = new StringBuilder();
        for(StringBuilder sb : list){
            result.append(sb.toString());
        }
        return result.toString();
    }
}

总结

本篇文章讲解了算法题目的思路和解法，代码和笔记由于纯手打，难免会有纰漏，如果发现错误的地方，请第一时间告诉我，这将是我进步的一个很重要的环节。以后会定期更新算法题目以及各种开发知识点，如果您觉得写得不错，不妨点个关注，谢谢。