[LeetCode] 6. Z 字形变换

题目链接:(https://leetcode-cn.com/problems/zigzag-conversion/)

题目描述:

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

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

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

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

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例:

示例 1:

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

示例 2:

输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:

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

思路:

1. 模拟过程

   Z字形,就是两种状态,一种垂直向下,还有一种斜向上

   控制好边界情况就可以了.

2. 找规律

如上图所示,我们发现规律

1. 每一个Z字的首字母差,numRows*2-2位置
2. 出去首尾两行,每个Z字有两个字母,索引号关系为,一个为i,另一个为numsRows*2-2-i

大家可以关注我的知乎专栏,了解更多解题技巧!

代码:

思路1:

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if not s:
            return ""
        if numRows == 1:return s
        s_Rows = [""] * numRows
        i  = 0
        n = len(s)
        while i < n:
            for j in range(numRows):
                if i < n:
                    s_Rows[j] += s[i]
                    i += 1
            for j in range(numRows-2,0,-1):
                if i < n:
                    s_Rows[j] += s[i]
                    i += 1
        return "".join(s_Rows)

思路2:

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if not s:
            return ""
        if numRows == 1:return s
        split_s_len = numRows * 2 - 2
        data = []
        n = len(s)
        
        for i in range(0, n,split_s_len):
            data.append(s[i:i+split_s_len])
        #print(data)
        res = ""
        for i in range(numRows):
            for tmp in data:
                if i < len(tmp):
                    if i == 0 or i == numRows-1:
                        res += tmp[i]
                    else:
                        res += tmp[i]
                        if split_s_len -i  < len(tmp):
                            res += tmp[split_s_len-i]
        return res

java

class Solution {
    public String convert(String s, int numRows) {
        //字符串转化成数组
        char[] c = s.toCharArray();
        //创建字符串数组
        StringBuffer[] sb = new StringBuffer[numRows];
        for(int i = 0; i < sb.length; i++)
            sb[i] = new StringBuffer();
        int i = 0;
        int len = c.length;
        while(i < len){
            for(int idx = 0; idx < numRows && i < len; idx++)
                sb[idx].append(c[i++]);
            for(int idx = numRows - 2; idx >= 1 && i < len; idx--)
                sb[idx].append(c[i++]);
        }
        //拼接所有字符串
        for(int idx = 1;idx < numRows;idx++)
            sb[0].append(sb[idx]);
        return sb[0]oString();
    }
}