ZigZag Conversion —— leetcode

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

本题要求我们将给定字符串按照给定的行数进行转换，转换方法如下图所示：

本题较为简单，只需要求出每个字母的数学表达式即可。根据计算可以发现，一个循环周期为2 *（nRow - 1），如第一行“P”和“A”相差了2*（3-1）= 4。

同时，除了第一行和最后一行，中间的行每个循环之间还有一个字母需要输出，如第二行的“A”和“L”之间还有一个“P”要输出，而该字母（“P”）与其前一个字母（“A”）之间相差2 *（nRow - i - 1），i为当前行数。

算出表达式候，即可写出程序，该算法复杂度为O(L)，L为字符串长度

该程序代码如下：

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