Java算法-力扣leetcode-6. Z 字形变换

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

解:

• 定义numRows长度的数组, 数组中存放不定长List
• 依次变量String每个char
• 找到规律 每2*numRows-2个元素为一个组

• 对2*numRows-2取余数
• 小于numRows时依次录入数组
• 大于numRows时依次倒着录入数组
• 等于0时录入数组中第二组

class Solution {
    public String convert(String s, int numRows) {
        ArrayList<String> r[] = new ArrayList[numRows];
        for (int i = 0; i < numRows; i++) {
            r[i] = new ArrayList<String>();
        }
        if(s.length() <= numRows){
            return s;
        }
        if(numRows==1){
             return s;
        }
        r[0].add("" + s.charAt(0));
        for (int i = 1; i < s.length(); i++) {
            int temp = (i + 1) % (2 * numRows - 2);
            if (temp == 0) {
                r[1].add("" + s.charAt(i));
            } else if (temp <= numRows) {
                r[temp - 1].add("" + s.charAt(i));
            } else {
                r[numRows - (temp - numRows) - 1].add("" + s.charAt(i));
            }
        }
        String returnString = "";
        for (List<String> list : r) {
            for (String st : list) {
                returnString += st;
            }
        }
        return returnString;
    }
}