力扣第六题-Z字形变换|Java 刷题打卡

264 阅读1分钟

本文正在参加「Java主题月 - Java 刷题打卡」,详情查看 活动链接

前言

力扣第六题如下所示:

在这里插入图片描述

一、思路

理解一下题目的意思就是:
将字符串以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天!
掘金币,我来啦😁