（LeetCode）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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zigzag-conversion
 

此题有两个思路，第一个：每次输入字符串S后，会先将他存入一个二维数组中，然后，一行数组一行数组的拼接。主要思路在于，从上往下，然后从左往右循环将数据存入二维数组：

class Solution {
    public String convert(String s, int numRows) {
        //初始化
        int row=0,col=0;
        String x = "";
        //如果行数为1或者行数大于字符串长度，则直接返回字符串
        if (numRows == 1 || s.length() <= numRows){
            return x = s;
        }
        //定义初始化二维数组
        char[][] z = new char[numRows][1000];
        for (int i = 0; i < s.length(); i++) {
            if (row == numRows){
                row--;
                while (row > 0 && i <s.length()){
                    row--;
                    col++;
                    z[row][col] = s.charAt(i);
                    i++;
                }
                //其实上面那个i++应该换成其他变量，然后在这一行赋值给i，但是既然通过了，就算了
                i = i - 2;
                continue;
            }
            z[row][col] = s.charAt(i);
            row++;
        }
        
        for (int i = 0; i < numRows; i++) {
            x = x + String.copyValueOf(z[i]);
        }
        //leetcode 会保存字符数组中为空的代码“\u0000”，因此需要将其替换
        x = x.replace("\u0000", "");  
        x = x.replace("\\u0000", ""); 
        return x;
    }
}

在这个地方会出现一个小问题，就是char[][] z = new char[numRows][1000];这一行代码，会给数组首先赋值\u0000，这个在ide里面返回的时候，不会出现，但是在LeetCode里面会报错，所以最后需要加两行代码，把\u0000替换为空。

第二个思路就是，直接上数学方法，字符串S，行数为row，0<=i<S.length()

0                4                8

1        3      5        7

2                6 

第一行的规律为   i%(2*row-2) == 0

第二行的规律为   i%(2*row-2) == 1 || 2n-2-1

.

.

.

最后一行的规律为 i%(2*row-2) == n-1 

代码

public static String convert(String s, int numRows) {

        if(numRows ==0 || numRows ==1){
            return s;
        }
        ArrayList<String>[] lists = new ArrayList[numRows];
        for (int i = 0; i < lists.length; i++) {
            lists[i] = new ArrayList<>();
        }
        for (int i = 0; i < s.length(); i++) {
            String c = s.charAt(i)+"";
            int target = i % (2 * numRows - 2);
            if(target>=numRows){
                target = (2 * numRows - 2) -target;
            }
            lists[target].add(c);
        }
        StringBuilder builder = new StringBuilder(s.length());
        for (ArrayList<String> list : lists) {
            for (String s1 : list) {
                builder.append(s1);
            }
        }
        return builder.toString();
    }