Leetcode 6. Z 字形变换

6. Z 字形变换

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，排列如下：

L   C   I   R
E T O E S I I G
E   D   H   N

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串

比如："LCIRETOESIIGEDHN"。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例 1:

输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"

示例 2:

输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:

L     D     R
E   O E   I I
E C   I H   N
T     S     G

实现代码

import java.util.LinkedList;

/**
 * https://leetcode-cn.com/problems/zigzag-conversion/
 * 6. Z 字形变换
 */
public class Problem6 {

    public static void main(String[] args) {
        // LEETCODEISHIRING PAYPALISHIRING
        String s = "PAYPALISHIRING";
        int numRows = 8;
        String result = convert(s,numRows);
        System.out.println(result);
    }

    public static String convert(String s, int numRows) {
        int length = s.length();
        if(numRows<2 || numRows>length) return s;
        char arr[][] = new char[numRows][length];
        char ch[] = s.toCharArray();
        int row = 0,col = 0;
        for (int i=0;i<length;i++) {
            // Z字一行
            if (col==0 || col%(numRows-1)==0){
                if(row<numRows-1){
                    arr[row++][col] = ch[i];
//                    System.out.println(arr[row++][col] = ch[i]);
                } else if(row==numRows-1){
                    arr[row--][col++] = ch[i];
//                    System.out.println(arr[row--][col++] = ch[i]);
                }
            } else if (row>0) { // Z字一折
                arr[row--][col++] = ch[i];
//                System.out.println(arr[row--][col++] = ch[i]);
            } else if (row==0) { // Z字一折
                arr[row++][col] = ch[i];
//                System.out.println(arr[row++][col] = ch[i]);
            }
        }
        StringBuilder sb = new StringBuilder();
        for(char[] arr1:arr){
            for(char arr2:arr1){
                if(arr2 != 0){
                    sb.append(String.valueOf(arr2));
                }
            }
        }
        return sb.toString();
    }
}