力扣第6题 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);

示例 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

解答

我在excel写了一张图

上面为总行数为3、4、5行时的一个例子

算法标题是Z形，实际大家看着可能是倒N形

从总行数为3来看

• 第一列是0-4-8-12
• 第三列为2-6-10-14

很容易发现他们之间都是在前一个基础上加4

再从总行数为4的来看

• 0-6-12
• 3-9-15

在前面一个数的基础上加6

总行数5的来看

• 0-8-16
• 4-12-20

在前面一个数的基础上加8

通过在和三例可以很容易发现，都是在前一个数的基础上加

2*(总行数-1)

我这里将他定义为line

int line = 2*(numRows-1);

那么第一行和最后一行便解决了

在来找其他几行的

总行数为3时：

• 1-3-5-7-9-11-13（当前行数为2）

总行数为4：

• 1-5-7-11-13（当前行数为2）
• 2-4-8-10-14（当前行数为3）

总行数为5：

• 1-7-9-15-17（当前行数为2）
• 2-6-10-14-18（当前行数为3）
• 3-5-11-13-19（当前行数为4）

直接看可能很难发现，那么便拆分开，我们把两个数中间的一个数拿出来单独看

总行数为3时：

当前行数为2：

• 1-5-9-13
• 3-7-11

总行数为4：

当前行数为2：

• 1-7-13
• 5-11

当前行数为3：

• 2-8-14
• 4-10

通过总行数为3和4的例子，其实可以发现

摘掉中间那个数后，发现他们之间的差是line，也就是2*(总行数-1)

可以发现

下面那行数=上面那个数+line-2*（当前行数-1）

而line=2*(总行数-1)

所以我们在循环时，一次循环保存两个数（上面和下面的数），循环一次结束后加line，开始下一次

先输出上面那行数，然后紧接着输出下面那行数

不多说，上代码

public String convert(String s, int numRows) {
        int line  = 2*(numRows-1);
		//如果分为一行或者字符串长度小于总行数，无法分配时直接返回当前字符串
        if(numRows==1||s.length()<numRows){
            return s;
        }
        StringBuffer s1  = new StringBuffer();
		//开始遍历，i为当前行数，这里从0开始，0即第一行
        for (int i = 0; i < numRows; i++) {
		//在当前行进行遍历，每次遍历结束后加Line
            for (int j = i; j < s.length();j+=line) {
			//如果是第一行或者最后一行，则直接加叠加到StringBuffer中
                if(i==0||i==numRows-1){
                    s1.append(s.charAt(j));
                }else {
				//如果不是最后一行，首先将上面的数叠加进去
                    s1.append(s.charAt(j));
					//然后将下面那个数叠加进去，如果下面的数索引没有超过数组长度则叠加、
                    if((j+line-2*i)<s.length()){
                        s1.append(s.charAt(j+line-2*i));
                    }
                }
            }
        }
        return s1.toString();
    }

这里注意的是，StringBuilder和StringBuffer以及String的区别

StringBuilder性能上会更好，但是线程不安全

StringBuffer线程安全，但性能比StringBuilder

String则性能最差，因为String不可变，每次更改都要生成新的String对象

这里我就用线程安全的StringBuffer为例，如果你想考虑性能快可以使用StringBuilder，最后是提交截图