leetcode 6.Z字形变换

leetcode 6.Z字形变换

题目描述

将一个给定字符串根据给定的行数，以从上往下、从左到右进行 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

解题思路

该题考察的是对字符串的操作，将一个给定字符串根据给定的行数，以从==上往下、从左到右==进行 Z 字形排列。如果觉得题目描述的不清楚，看一下下面的动图¹就可以很清楚的明白是怎么进行Z字形变形的啦。知道了转换的过程，按照上面的思路转换成代码就可以了。在这里我们使用的是建立一个二维的vector，更加变形规则不断的向每一行添加字符，最后再把每一个vector中的字符转成一整个字符串。代码如下：

class Solution {
public:
    string convert(string s, int numRows) {
        // 如果s是空，或者说numRows的数字小于等于1，就表示不需要转换，直接返回原始字符串
        if(numRows <=1 || s.size() == 0){
            return s;
        }
        
        vector<vector<char>> arr(numRows, vector<char>(0,0)); // 建立一个二维数组，初始化为空
        string res;	      // 存储转换后的字符串
        int len = s.size();      // 原始字符串的长度
        bool flag = false;       // 定义一个标志位，用于z行排列的变化
        int row = 0;			// 定义一个记录行号的变量，在该行下添加字符
        for(int i=0; i<len; i++){
            if(!flag){     // 如果flag为false，表示要从上到下添加字符
                if(row == numRows-1){ // 如果row走到了最后一行，flag状态翻转，下次进行从左到右添加
                    flag = true;    
                    arr[row].push_back(s[i]);
                    row--;
                }
                else{
                    arr[row].push_back(s[i]);
                    row++;
                }
            }
            else{     // 如果flag为true，表示要从左到右添加字符
                if(row == 0){        // 如果row走到了第0行，flag翻转，下次进行从上到下添加
                    flag = false;      
                    arr[row].push_back(s[i]);
                    row++;
                }
                else{
                    arr[row].push_back(s[i]);
                    row--;
                }
            }
        }
        // 这个数组完成添加以后，逐行的合并字符串。
        for(int i=0; i<numRows; i++){
            for(int j=0; j<arr[i].size(); j++){
                res += arr[i][j];
            }
        }
        
        return res;
    }
};

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分享技术问题，我们一起学习进步

---

1. https://leetcode-cn.com/problems/zigzag-conversion/solution/zzi-xing-bian-huan-by-jyd/ ↩︎