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 (s.isEmpty() || numRows <= 1) {
            return s;
        } else {
            List<Cell> cells = new ArrayList<>(s.length());
            Cell cell = new Cell(s.charAt(0), 0, 0);
            cells.add(cell);
            boolean down = true;
            for (int i = 1; i < s.length(); i++) {
                if (down ? (cell.getRow() + 1 == numRows) : (cell.getRow() == 0)) {
                    down = !down;
                }
                if (down) {
                    cell = new Cell(s.charAt(i), cell.getRow() + 1, cell.getColumn());
                } else {
                    cell = new Cell(s.charAt(i), cell.getRow() - 1, cell.getColumn() + 1);
                }
                cells.add(cell);
            }
            cells.sort(Cell::compareByPosition);
            StringBuilder sb = new StringBuilder(s.length());
            cells.forEach(p -> sb.append(p.getChar()));
            return sb.toString();
        }
    }

    private static class Cell {
        private char ch;
        private int row;
        private int column;

        public Cell(char ch, int row, int column) {
            this.ch = ch;
            this.row = row;
            this.column = column;
        }

        public char getChar() {
            return this.ch;
        }

        public int getRow() {
            return this.row;
        }

        public int getColumn() {
            return this.column;
        }

        public static int compareByPosition(Cell c1, Cell c2) {
            int ret = c1.getRow() - c2.getRow();
            if (ret == 0) {
                ret = c1.getColumn() - c2.getColumn();
            }
            return ret;
        }
    }
}