Z字形变换 力扣T6

Z字形变化

目录

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);

示例：

示例一：

输入：s = "PAYPALISHIRING", numRows = 3
输出："PAHNAPLSIIGYIR"

示例二：

输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"
解释：
P     I    N
A   L S  I G
Y A   H R
P     I

示例三：

输入：s = "A", numRows = 1
输出："A"

代码：

class Solution {
    public String convert(String s, int numRows) {
        // 分别按行开始
        if (numRows == 1) return s;
        StringBuffer str = new StringBuffer();
        for (int index = 0; index < numRows; index ++) {
            int count = (numRows - 1)*2, i = index; 
            // 行数从 0 开始
            // Z字形第一行和最后一行都是差相同的数 即 （总的行数 - 1）*2
            // 其余行的相差数都由两个组成 分别为 （总的行数-当前行 - 1 ）*2 和 2*当前行数
            // 两个间隔数顺序不能改变 且 间隔分布
            if (index != 0 && index != numRows - 1) {
                int count1 = (numRows - index - 1)*2;
                int count2 = 2*index;
                boolean sign = true; 
                // 设置一个sign来表示间隔分布
                // 如 true代表第一个间隔数 false代表第二个间隔数
                while (i < s.length()) {
                    str.append(s.charAt(i));
                    if (sign) {
                        i += count1;
                        sign = false;
                    } else {
                        i += count2;
                        sign = true;
                    }
                }
            } else {
                while (i < s.length()) {
                    str.append(s.charAt(i));
                    i += count;
                }
            }
        }
        return str.toString();
    }
}