LeetCode 6. Z字形变换（ZigZag Conversion）

题目描述

 

将字符串 "PAYPALISHIRING" 以Z字形排列成给定的行数：

P   A   H   N
A P L S I I G
Y   I   R

之后从左往右，逐行读取字符："PAHNAPLSIIGYIR"

实现一个将字符串进行指定行数变换的函数:

string convert(string s, int numRows);

示例 1:

输入: s = "PAYPALISHIRING", numRows = 3
输出: "PAHNAPLSIIGYIR"

示例 2:

输入: s = "PAYPALISHIRING", numRows = 4
输出: "PINALSIGYAHRPI"
解释:

P     I    N
A   L S  I G
Y A   H R
P     I

 

解题思路

 

把字符串按行输出时要考虑首尾与中间情况，所以首先计算出每个Z字形周期内的字符个数，然后求出周期数，接着对每一行的每个周期依次遍历，对于首尾行的字符直接添加；对于中间行的字符要再添加Z字中间的字符。对于最后一个周期，要判断是否走到了字符串末尾。

 

代码

 

class Solution {
public:
    string convert(string s, int numRows) {
        if(numRows == 1) return s;
        int cycle = 2 * numRows - 2;
        int numZ = s.length() / cycle;
        string res = "";
        for(int i = 0; i < numRows; i++){
            for(int j = 0; j <= numZ; j++){
                if(j * cycle + i >= s.length()) continue;
                if(i == 0 || i == numRows - 1)
                    res += s[j * cycle + i];
                else{
                    res += s[j * cycle + i];
                    int idx = j * cycle + cycle - i;
                    if(idx < s.length())
                        res += s[idx];
                }
            }
        }
        return res;
    }
};