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

二、代码

class Solution {
public:
    string convert(string s, int numRows) {
    string str="";
    if(numRows==1){
        str = s;
    }else{
        for(int line=0; line<numRows; line++){
            for(int i=line; i<s.size(); i+=2*numRows-2){
                str += s[i];
                if(line>0 && line<numRows-1 && i+2*(numRows-1-line)<s.size()){
                    str += s[i+2*(numRows-1-line)];
                }
            }   
        }   
    }

    return str;
    }
};