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

    }

}