【Leetcode】【Easy】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".

 

解题：

引用http://blog.csdn.net/zhouworld16/article/details/14121477的图片说明：

所谓的ZigZag，就是这样的：

因此可以找到每行排列数字的规律，nRows = n，字符串字符下标从0开始。

 

普通青年：

①   先读第1行：0， 0 + 2n-2， 0 + (2n-2)*2，...

②   读第2行至第n-2行，设行数为e，则每行读取数字为：

　　e-1，e-1 + 2n-2 - (e-1)*2， e-1 + 2n-2， e-1+2n-2 + 2n-2 - (e-1)*2， ... 

　　规律即，两个数字一组，每组间隔2n-2：

　　for (int i=e-1; i<string.length(); i+=2n-2) {

　　　　每组元素可表示为i 和 i + 2n-2 - (e-1)*2

　　}

③  读最后一行：n-1， n-1 + 2n-2， n-1 + (2n-2)*2， ...

（由于每循环一组，会读出两个数字。注意判定字符串越界）

class Solution {
public:
    string convert(string s, int nRows) {
        string res;
        int len = s.length();
        
        if (nRows == 0)
            return res;
            
        if (nRows == 1)
            return s;
            
        for(int i=0; i<len; i+=2*nRows-2)
            res.push_back(s[i]);
        
        int index = 1;
        while(nRows-1-index){
            for(int i=index; i<len; i+=2*nRows-2) {
                res.push_back(s[i]);
                int zig = i + 2 * nRows - 2 * index - 2;
                if (zig < len)
                    res.push_back(s[zig]);
            }
            index++;
        }
        
        for(int i=nRows-1; i<len; i+=2*nRows-2)
            res.push_back(s[i]);
        
        return res;

    }
};

 

聪明的青年：

类似于期末考试考场排座位，一般为蛇形排号。将考场的座位按一竖列为一组，编号时，第一组的座位从前向后编号，对第二组从后向前进行编号，如下图：

① ⑧ ⑨

② ⑦       ...

③ ⑥

④ ⑤

nRows即为每组能坐多少人。因此先让同学们按编号入座，再从前向后按横排依次读出他们的编号即为本题答案。

class Solution {
public:
    string convert(string s, int nRows) {
        string sArray[nRows];
        string res = "";
        int index = 0;
        
        if (nRows == 1)
            return s;
        
        for(int i=0; i<s.length(); ++i){
            sArray[index] += s[i];
            
            if (!(i / (nRows-1) % 2)) {
                index++;
            } else {
                index--;
            }
        }

        for(int i=0; i<nRows; ++i){
            res += sArray[i];
        }

        return res;
    }
};

 

当然是原创的。