題目

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

分析

這個題目十分有意思。它把字元換進行N字形排列然後一行一行輸出。作為一個normal題肯定不可能是構造二維陣列然後進行遍歷。實際上仔細分析一下轉換後的結果就會發現：主列（例如Example2 裡面PAYP那一列與ISHI那一列和NG那一列）之間的下標差值都是固定的，即(2*numRows)-2。然後除了第一行和最後一行之外，其他行的構造都是主列、非主列(即N字形斜著的那部分)、主列、非主列、主列、非主列……然後非主列元素的下標都可以根據主列元素的下標找到。比如第2行，主列元素下標-2就是非主列元素下標，然後第3行主列元素下標-4就是非主列元素……

程式碼

class Solution {
public:
    string convert(string s, int nRows) {
        if (nRows <= 1) return s;
        string res = "";
        int size = 2 * nRows - 2;
        for (int i = 0; i < nRows; ++i) {
            for (int j = i; j < s.size(); j += size) {
                res += s[j];
                int tmp = j + size - 2 * i;
                if (i != 0 && i != nRows - 1 && tmp < s.size()) res += s[tmp];
            }
        }
        return res;
    }
};