LeetCode No.6 Z 字形变换

题目描述

本题至少应该想到用一个二维数组存储结果，但是这样会造成很多空间浪费。

string convert(string s, int numRows) {

	if (numRows == 1) return s; //特例判断，非常重要

	vector<string> rows(min(numRows, int(s.size()))); //vector保存每一行的字符串
	int curRow = 0;
	bool goingDown = false; //指定row应该向下增加还是向上减少

	for (char c : s) {
		rows[curRow] += c;
		//到了0行或者最后一行要调头
		if (curRow == 0 || curRow == numRows - 1) goingDown = !goingDown; 
		curRow += goingDown ? 1 : -1;
	}

	string ret;
	for (string row : rows) ret += row;
	return ret;
}