Java LeetCode题 6 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

代码

class Solution {
    public String convert(String s, int numRows) {
        if (numRows == 1) {
            return s;
        }
        numRows -= 1;
        char[][] charsArr;
        int strLen = s.length();
        if (strLen % numRows > 0) {
            charsArr = new char[(strLen / numRows) + 1][];
        } else {
            charsArr = new char[strLen / numRows][];
        }
        int i = 0;
        for (; (i + 1) * numRows <= strLen; i++) {
            charsArr[i] = s.substring(i * numRows, (i + 1) * numRows).toCharArray();
        }
        if (strLen > i * numRows && strLen < (i + 1) * numRows) {
            charsArr[i] = s.substring(i * numRows, strLen).toCharArray();
        }
        StringBuilder sb = new StringBuilder();
        for (int j = 0; j <= numRows; j++) {
            for (int n = 0; n < charsArr.length; n++) {
                char[] arr = charsArr[n];
                if (n % 2 == 0) {
                    if (j < arr.length) {
                        sb.append(arr[j]);
                    }
                } else {
                    if (numRows - j < arr.length) {
                        sb.append(arr[numRows - j]);
                    }
                }
            }
        }
        return sb.toString();
    }
}

思路讲解

首先，我的思路是将这些z型变换的概念，转换成一对对由升序的竖字符串和一个倒叙的斜字符串组合而成的图形。且在最后拼接字符串时，每个竖字符串所取的位置是一样的，斜字符串也一样。所以我这里用一个二维的char数组的第一维来存储这些字符串，第一维的偶数下标代表竖字符串，奇数下标代表斜字符串（下标从0开始）。

运行结果

info
解答成功:
执行耗时:4 ms,击败了84.84% 的Java用户
内存消耗:38.5 MB,击败了92.10% 的Java用户