【LeetCode】6. ZigZag Conversion（C++）

地址：https://leetcode.com/problems/longest-palindromic-substring/

题目：

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)

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:

理解：

把给定的字符串按zigzag的形式写出来，然后按行输出。下面是我的实现。
其中2 * numRows - 2 - 2 * i可以指向某列右边的元素，但是对最后一行而言，更新的结果为0也就是指向自己，因此需要跳过；
在斜边上，还需要跳第二次到右侧的直边，这种情况下由于第一行上次就已经到了，因此不用考虑第一行。
也就是下图红线和蓝线的部分。

实现：

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

看到一种有趣的实现，不用考虑下标，而是用一个string数组来保存每次的行，最后把这个数组合并。

class Solution {
public:
	string convert(string s, int nRows) {
		if (nRows <= 1)
			return s;
		string *str = new string[nRows];

		int row = 0, step = 1;
		for (int i = 0; i < s.size(); ++i)
		{
			str[row].push_back(s[i]);
			if (row == 0)
				step = 1;
			else if (row == nRows - 1)
				step = -1;
			row += step;
		}
		string res;
		for (int j = 0; j < nRows; ++j)
		{
			res.append(str[j]);
		}
		delete[] str;
		return res;
	}
};