ZigZag Conversion -- leetcode

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" .

ZigZag的一个形象图解：（图片来源）

class Solution {
public:
    string convert(string s, int nRows) {
        if(nRows <= 1) return s; // 省略此句，会导致死循环，当nRows == 1时
        string result;
        result.reserve(s.size());
        const int segment = nRows * 2 - 2;
        for (int i=0; i<nRows; i++) {
                for (int j=i; j<s.size(); j+=segment) {
                        result += s[j];
                        if (i!=0 && i!=(nRows-1) && (j+segment-2*i)<s.size())
                                result += s[j+segment-2*i];
                }
        }
        return result;
    }
};