leetcode题库——Z字形变换

题目描述：

将字符串 "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

方法：

class Solution {
public:
    string convert(string s, int numRows){
        string str;
        int numCols,inGroupNum,groupNum,res;
        if(s.size()==0) return str="";//空串
        if(numRows==1) return s;//输出一行
        if(numRows==2){inGroupNum=2;}//输出两行，每组几个数
        else inGroupNum=2*numRows-2;//多于两行

        if(s.size()%inGroupNum==0) groupNum=s.size()/inGroupNum;
        else groupNum=s.size()/inGroupNum+1;//有几组
        
        int rest=s.size()-(groupNum-1)*inGroupNum;//最后一组几个数
        if(rest<=numRows) res=1;//最后一组的数能分到几列
        else res=rest-numRows+1;
        
        numCols=inGroupNum*(groupNum-1)+res;//数组应分配的列数
        char Z[numRows][numCols];
        
        for(int i=0;i=numRows) row--;
                }    
                else break;
            }
            
            for(int k=0;k

思路：

将Z字形字符串分割成group，如下图：

设一个字符数组用于接收排列成Z字形的字符，并计算好group数，每个group内有几个字符等信息，并且需要考虑空串，以及numRows为1和2的特殊情况。