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

---

題目的意思是把字符串上下上下走之字形狀,然后按行輸出,比如包含數字0~22的字符串, 給定行數為5,走之字形如下:

現在要按行輸出字符,即:0 8 16 1 7 9 15 17 2…….

如果把以上的數字字符看做是字符在原數組的下標, 給定行數為n = 5, 可以發現以下規律:

(1)第一行和最后一行下標間隔都是interval = n*2-2 = 8 ;                                                   本文地址

(2)中間行的間隔是周期性的,第i行的間隔是: interval–2*i,  2*i,  interval–2*i, 2*i, interval–2*i, 2*i, …

 

代碼如下，時間復雜度為O（n），n是字符串的長度：

class Solution {
public:
    string convert(string s, int nRows) {
        if(nRows == 1)return s;
        int len = s.size(), k = 0, interval = (nRows<<1)-2;
        string res(len, ' ');
        for(int j = 0; j < len ; j += interval)//處理第一行
            res[k++] = s[j];
        for(int i = 1; i < nRows-1; i++)//處理中間行
        {
            int inter = (i<<1);
            for(int j = i; j < len; j += inter)
            {
                res[k++] = s[j];
                inter = interval - inter;
            }
        }
        for(int j = nRows-1; j < len ; j += interval)//處理最后一行
            res[k++] = s[j];
        return res;
    }
};

 

【版權聲明】轉載請注明出處：http://www.cnblogs.com/TenosDoIt/p/3738693.html