6 ZigZag Conversion

1 题目

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 s, int numRows);

Example 1:

Input: s = "PAYPALISHIRING", numRows = 3
Output: "PAHNAPLSIIGYIR"

Example 2:

Input: s = "PAYPALISHIRING", numRows = 4
Output: "PINALSIGYAHRPI"
Explanation:

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

2 尝试解

class Solution {
public:
    string convert(string s, int numRows) {
        if(numRows < 2) return s;
        string result = "";
        for(int i = 0; i < numRows; i++){  //cout line by line
            result = result + s[i];
            for(int j = i; j < s.size(); j += 2*(numRows-1)){  
                if(i!=0 && i != numRows-1 && j + 2*(numRows-i-1) < s.size()){
                    result = result + s[j+2*(numRows-i-1)];  //第二行A到L距离为4，第三行Y到A距离为2
                }
                if(j+2*(numRows-1) < s.size()){
                    result = result + s[j+2*(numRows-1)];//第一行P到I距离为6，第二行A到S距离为6   
                }
            }
        }
        return result;
    }
};

3 标准解

class Solution {
public:
    string convert(string s, int numRows) {

        if (numRows == 1) return s;

        vector<string> rows(min(numRows, int(s.size())));
        int curRow = 0;
        bool goingDown = false;

        for (char c : s) {
            rows[curRow] += c;
            if (curRow == 0 || curRow == numRows - 1) goingDown = !goingDown;
            curRow += goingDown ? 1 : -1;
        }

        string ret;
        for (string row : rows) ret += row;
        return ret;
    }
};