leetcode-ZigZag Conversion

题目描述：

The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

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

And then read line by line: "PAHNAPLSIIGYIR"

Write the code that will take a string and make this conversion given a number of rows:

string convert(string text, int nRows);

convert("PAYPALISHIRING", 3) should return "PAHNAPLSIIGYIR".

这题网上有不少解法，下面是我自己的一种解法：

思想：把zigZig字符串分成一段段，段长是partslen=nRos*2-2。每个段是一个竖线和斜线组成。那么转换成字符串时，分成2种情况，首尾行和中间行

设i是行号，j是段号，则有

首尾行元素对应坐标:j*partslen+i

中间行元素除了包含上面这个坐标的元素外，还要有(j+1)*partslen-i，如图中‘P’的坐标就是2*2-1，当然需要有一些边界判断。

代码如下：

class Solution {
public:
    string convert(string s, int numRows) {
        if(s.size()<=1||numRows<=1)return s;
        int partslen=2*numRows-2;
        int parts=s.size()/(2*numRows-2);
        string result;
        for(int i=0;i<numRows;++i)
        {
            for(int j=0;j<=parts;++j)
            {
                
                if((i==0||i==numRows-1)&&(j*partslen+i)<s.size())
                {
                    result.push_back(s[j*partslen+i]);
                }
                else if((j*partslen+i)<s.size())
                {
                    result.push_back(s[j*partslen+i]);
                    if((j+1)*partslen-i<s.size())
                    result.push_back(s[(j+1)*partslen-i]);
                }
            }
            
        }
        return result;
    }
};