力扣 Z 字形变换 C++

扁扁熊思路图解：

代码：

/*
将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时，排列如下：

L   C   I   R
E T O E S I I G
E   D   H   N
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："LCIRETOESIIGEDHN"。

示例 1:

输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"

示例 2:

输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"

*/

#include <vector>
using namespace std;

#define max(a,b) (((a) > (b)) ? (a) : (b))
#define min(a,b) (((a) < (b)) ? (a) : (b))

class Solution {
public:
	string convert(string s, int numRows) {
		if (numRows == 1) return s;

		string ret;
		vector<string> rows(min(numRows, int(s.size())));//档位长度
		int curRow = 0;//当前档位
		bool flag = false;

		for (char c : s)
		{
			rows[curRow] += c;//哈希累加
			if (curRow == 0 || curRow == numRows - 1)//转弯 当前行curRow为0或numRows -1时，箭头发生反向转折
			{
				flag = !flag;
			}
			curRow += flag ? 1 : -1;//档位
		}

		for (string row : rows)//连接所有的串
		{
			ret += row;
		}
		return ret;
	}
};


int main()
{
	string str = "LEETCODEISHIRING";

	Solution solution;
	string ret = solution.convert(str, 4);

	printf("ret ---- %s", ret.c_str());
	getchar();

    return 0;
}