Leetcode 6. ZigZag Conversion

最新推荐文章于 2021-03-20 20:07:40 发布

原创最新推荐文章于 2021-03-20 20:07:40 发布 · 267 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode

Leetcode 同时被 2 个专栏收录

409 篇文章

订阅专栏

Algorithm

260 篇文章

订阅专栏

本文介绍了如何实现一种特殊的字符串转换方法，即将输入的字符串以Z字形方式排列后重新组合输出。提供了两种版本的C++实现方案，通过调整变量追踪当前字符的位置和方向来完成转换。

文章作者：Tyan
博客：noahsnail.com | 优快云 | 简书

1. Description

ZigZag Conversion

2. Solution

Version 1

class Solution {
public:
    string convert(string s, int numRows) {
        if(numRows == 1 || s.length() <= numRows) {
            return s;
        }
        string result;
        vector<string> lines(numRows);
        int row = -1;
        bool flag = true;
        for(char ch : s) {
            row = flag?++row:--row;
            lines[row] += ch;
            if(row == 0) {
                flag = true;
            }
            else if(row == numRows - 1) {
                flag = false;
            }
        }
        for(string line : lines) {
            result += line;
        }
        return result;
    }
};

Version 2

class Solution {
public:
    string convert(string s, int numRows) {
        if(numRows == 1 || s.length() <= numRows) {
            return s;
        }
        string result;
        vector<string> lines(numRows);
        int row = -1;
        int step = 1;
        for(char ch : s) {
            row += step;
            lines[row] += ch;
            if(row == 0) {
                step = 1;
            }
            else if(row == numRows - 1) {
                step = -1;
            }
        }
        for(string line : lines) {
            result += line;
        }
        return result;
    }
};