leetCode:Z字形变换

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

思路：

通过观察可以看出来当numRows为3的时候第一排各个字符在原字符串中间隔为3，当numRow为4，间隔为5，当numRow为5，间隔为7，当numRow为6，间隔为9,，可看出规律。
并且随后n行分别是收个字符两边n个字符，可以得到规律。

代码：

class Solution {
    public String convert(String s, int numRows) {
        // 1.首先进行特殊情况的判断
        if (s.length() <= numRows) {
            return s;
        }
        if (numRows == 1) {
            return s;
        }
        //2.根据numRows算出第一排数字的间隔。
        //对应关系：
        //   numRows       第一排字母间隔
        //     3               3
        //     4               5
        //     5               7
        //     6               9
        //     7               11
        int interval = numRows * 2 - 3;

        //3.获取传入字符串长度。
        int len = s.length();

        //3.判断如果字符串长度在中间上升时则会造成丢失问题。
        //解决方法，对原有字符串进行添加长度，最后结果去除拼接的。
        //flush用来判断是否进行了拼接操作。
        StringBuilder handleStr = new StringBuilder(s);
        boolean flush = true;
        if (len % (interval + 1) == 0) {
            flush = false;
            handleStr.append("-");
        } else if (len % (interval + 1) >= numRows) {
            flush = false;
            for (int i = 0; i < numRows; i++) {
                handleStr.append("-");
            }
        }
        //4.处理完成后对原有字符串进行覆盖。
        s = handleStr.toString();
        //5.主要逻辑对字符串进行处理。
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < numRows; i++) {
            int j = 0;
            do {
                if (i == 0 && j == 0) {
                    sb.append(s.charAt(0));
                } else if (i == (numRows - 1)) {
                    if ((j + i) < s.length()) {
                        sb.append(s.charAt(j + i));
                    }
                } else {
                    if (j - i > 0 && (j - i) != (j + i)) {
                        sb.append(s.charAt(j - i));
                    }
                    if (j + i < s.length() && (j - i) != (j + i)) {
                        sb.append(s.charAt(j + i));
                    }
                    if ((j - i) == (j + i) && (j - i) < s.length() && (j - i) >= 0) {
                        sb.append(s.charAt(j - i));
                    }
                }
                j = j + interval + 1;
            } while (j <= s.length());
        }

        //6.如果没有进行拼接操作则直接返回，否则循环去除字符串中的"-"
        if (flush) {
            return sb.toString();
        } else {
            String str = sb.toString();
            StringBuilder convertStr = new StringBuilder();
            for (int i = 0; i < str.length(); i++) {
                if (str.charAt(i) != '-') {
                    convertStr.append(str.charAt(i));
                }
            }
            return convertStr.toString();
        }
    }
}