LeetCode OJ : 6 ZigZag Conversion

readyao發表於2015-12-17

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

這個例子分為了三行,所以需要三個string物件來儲存每一行的字串,最後按順序(從第一排到最後一排)將字串相加起來;這個求和之後的字串就是所求,將它返回;

把變換之後的看成是一個週期變化的,第一個週期 P-A-Y-P,第二個週期A-L-I-S,第三個週期....

每個週期又有規律可循,P-A-Y分別存放在0-1-2行;接下來的p存放在1行;然後又是下一個週期。。。。


比如是四排的時候;


第一個週期A-B-C-D-E-F;首先將A-B-C-D存放在0-1-2-3排,再將E-F存放在2-1排。。所以存放在不同排的順序應該是0-1-2-3-2-1;

所以A-B-C-D-E-F對應存放在0-1-2-3-2-1;

每2 * numRows-2個字元依次存放在0-1-2-....-(numRows-1)-(numRows-2)-......--1;

class Solution {
public:
    string convert(string s, int numRows) {
        vector<string> vstr(numRows);
        int i = 0, j = 0;
        
        if(s.size() == 0){
            return s;
        }
        
        for (i = 0; i < s.size(); ){
            for (j = 0; j < numRows; ++j){
                if(i == s.size()){
                    break;
                }
                vstr[j].push_back(s[i++]);
            }
            
            for (j = numRows-2; j >=1 ; --j){
                if(i == s.size()){
                    break;
                }
                vstr[j].push_back(s[i++]);
            }
            
            if(i == s.size()){
                string str;
                for (i = 0; i < numRows; ++i){
                    str += vstr[i];
                }
                return str;
            }
        }
    }
};


執行效率比較低,不過這個執行時間不穩定,第一次提交的時候是22/%。。

相關文章