力扣第六题-Z字形变换

前言

力扣第六题如下所示：

一、思路

理解一下题目的意思就是：
将字符串以N的形式排列得到一个新的二维数组，再将二维数组输出为字符串

例子如下所示：

综上思路可以分为两个部分

• 从左到右遍历字符串，并将当前的元素插入到对应的行列（如上图中的第5个元素应为第三行的第二列）
• 遍历排列好的数组，输出结果即可

二、实现

代码实现

变量说明：
increase：是否增加
index：第几行
tips：在Java中可以使用StringBuilder提高性能

    public String convert(String s, int numRows) {
        if (s == null || numRows == 1)
            return s;
        StringBuilder[] builders = new StringBuilder[numRows];
        char[] chars = s.toCharArray();
        int index = 0;
        boolean increase = true;
        for (char c : chars) {
            StringBuilder buf = builders[index];
            if (buf == null)
                buf = new StringBuilder();
            builders[index] = buf.append(c);
            if (increase) {
                if (index >= numRows - 1) {
                    // 此时需减少
                    index--;
                    increase = false;
                } else {
                    index++;
                }
            } else {
                if (index <= 0) {
                    // 此时需增加
                    index++;
                    increase = true;
                } else {
                    index--;
                }
            }
        }
        StringBuilder ret = new StringBuilder();
        for (StringBuilder sb : builders) {
            if (sb != null)
                ret.append(sb);
        }
        return ret.toString();
    }

测试代码

    public static void main(String[] args) {
        Number6 n = new Number6();
        String ret = n.convert("LEETCODEISHIRING", 4);
        System.out.println(ret);
    }

三、总结

今天是力扣第六题~
本系列将会更新力扣的1-10题，连续更新10天！