LeetCode探索之旅2（90）-6ZigZag conversion

最新推荐文章于 2024-10-24 22:31:46 发布

JerryZengZ

最新推荐文章于 2024-10-24 22:31:46 发布

阅读量125

点赞数

CC 4.0 BY-SA版权

分类专栏：代码训练文章标签： LeetCode 找规律

本文链接：https://blog.youkuaiyun.com/JerryZengZ/article/details/90370819

代码训练专栏收录该内容

177 篇文章

订阅专栏

博客记录刷LeetCode第6题，将字符串用之字型表示。通过找规律发现每次叠加的下标与行数有关，提出用Python实现的问题并需注意下标，还附上了C++和Python代码。

今天继续刷LeetCode，第6题，将字符串用之字型表示。

分析：
通过找规律，发现每次叠加的下标与行数有关。

问题：
1、Python实现；
2、注意下标；

附上C++代码：

class Solution {
public:
    string convert(string s, int numRows) {
        string ret(s.length(), ' ');
        if (numRows > 1)
        {
                int step = (numRows - 1) * 2;
                int counter = 0;
                for (int j = 0; j < numRows; j++)
                {
                    for (int i = j; i < s.length(); i = i + step)
                    {
                        ret[counter] = s[i];
                        ++counter;
                        int offset = step - j * 2;
                        if (offset > 0 && offset < step)
                        {
                            int pos = offset + i;
                            if (pos < s.length())
                            {
                                ret[counter] = s[pos];
                                ++counter;
                            }
                        }
                    }
                 }
        } else {
                ret = s;
        }
        return ret;
    }
};

附上Python代码：

class Solution:
    def convert(self, s: str, numRows: int) -> str:
        if numRows == 1:
             return s
        zigzag = ['' for i in range(numRows)]  # 初始化zigzag为['','','']
        row = 0                                # 当前的行数
        step = 1                               # 步数：控制数据的输入
        for c in s:
            if row == 0:
                step = 1
            if row == numRows - 1:
                step = -1
            zigzag[row] += c
            row += step
        return ''.join(zigzag)