leetcode_p6_ZigZag_Conversion_c++

狮子HH

于 2016-10-24 11:50:18 发布

阅读量506

点赞数

分类专栏： Coding 文章标签： leetcode zigzag c++

本文链接：https://blog.csdn.net/yingyujianmo/article/details/52910322

版权

Coding 专栏收录该内容

32 篇文章 2 订阅

订阅专栏

本文介绍了LeetCode的第6题——Z字形变换，给出了解题思路和C++代码实现。题目要求根据给定的行数numRows，将字符串s以Z字形排列并返回变换后的字符串。通过分析示例，发现了与行数相关的规律，即步长与行数的关系，并据此编写了代码。

摘要由CSDN通过智能技术生成

leetcode p6: ZigZag_Conversion

给定一个字符串s和行数numRows, 先将该字符串按z形排成numRows行，然后按行输出所有字符组成的新字符串。

举个栗子：
zigzag conversion

从中发现：

这些数字都与6有关，称为magicNum，该数与行数有关，举个不同行的例子，可以发现规律： magicNum = 2*(numRows - 1) ;
对于每行的第一个数字，是以2为步长依次递减的，当为0时，重新赋值为magicNum;
对同一行，已知一个步长，它的下一步步长为magicNum减去当前步长。

c++ 代码如下

class Solution {
public:
    string convert(string s, int numRows) {
        if (numRows == 1)
            return s;
        string res = "";
        int magicNum = 2*(numRows - 1);
        for (int row = 0; row < numRows; ++row) {
            int initRowStep = magicNum - row*2; // 每行的起始步长
            if (initRowStep == 0) {             // 步长为0，重置为magicNum
                initRowStep = magicNum;
            }
            int step = initRowStep;
            for (int idx = row; idx < s.size(); ) {
                res += s[idx];
                idx += step;
                step = magicNum - step;
                if (step == 0) {
                    step = magicNum;
                }
            }
        }
        return res;
    }
};