LeetCode-(Z字形变换)

20200326

继续刷题

题目 ：Z字形变换

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

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

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

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

示例：

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

code1

class Solution {
    public String convert(String s, int numRows) {
		if(s==null || numRows<=1){
            return s;
        }
        int n = s.length();
        //取numRows，字符串s长度的较小值，用这个较小值来创建n个StringBuilder
        int rows = Math.min(n,numRows);
        StringBuilder[] arr = new StringBuilder[rows];
        for(int i=0;i<arr.length;++i){
            arr[i] = new StringBuilder();
        }
        int j=0;
        //向上还是向下的标志位
        boolean isDown = false;
        StringBuilder res = arr[0];
        //遍历字符串s，并将字符放到arr[j]中，之后根据标志位来判断往下还是往上
        for(int i=0;i<n;++i){
            arr[j].append(s.charAt(i));
            if(j==0 || j==numRows-1){
                isDown = !isDown;
            }
            if(isDown){
                ++j;
            }else{
                --j
            }
        }
        for(int i=1;i<arr.length;++i){
            res.append(arr[i]);
        }
        return res.toString();
    }
}

code2

class Solution {
    public String convert(String s, int numRows) {

        if (numRows == 1) return s;

        List<StringBuilder> rows = new ArrayList<>();
        for (int i = 0; i < Math.min(numRows, s.length()); i++)
            rows.add(new StringBuilder());

        int curRow = 0;
        boolean goingDown = false;

        for (char c : s.toCharArray()) {
            rows.get(curRow).append(c);
            if (curRow == 0 || curRow == numRows - 1) goingDown = !goingDown;
            curRow += goingDown ? 1 : -1;
        }

        StringBuilder ret = new StringBuilder();
        for (StringBuilder row : rows) ret.append(row);
        return ret.toString();
    }
}