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 == ) return s;
         int cycle =  * numRows - ;
         int numZ = s.length() / cycle;
         string res = "";
         for(int i = ; i < numRows; i++){
             for(int j = ; j <= numZ; j++){
                 if(j * cycle + i >= s.length()) continue;
                 if(i ==  || i == numRows - )
                     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;
     }
 };