LeetCode热门100之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);

代码：

class Solution {
public:
    string convert(string s, int numRows) {
         if(numRows <= 1 || s.size() <= numRows)
            return s;
        string ss;
        int len = s.size();

        for(int i = 0; i < numRows; i ++){
            if(i >= len) break;
            int j = i;
           
            if(i == 0 || i == numRows - 1){
                while(j < len){
                    ss += s[j];
                    j = j + 2*numRows-2;
                }
            }
            else 
                while(j < len){
                    ss += s[j];
                    if(j+2*(numRows-i-1) < len)
                        ss += s[j+2*(numRows-i-1)];
                    j = j + 2*(numRows-1);
                }    
        }
        return ss;
    }
};