LeetCode-----第六题-----Z 字形变换

Z 字形变换

难度：中等

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

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

L   C   I   R
E T O E S I I G
E   D   H   N

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："LCIRETOESIIGEDHN"。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例 1:

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

示例 2:

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

L     D     R
E   O E   I I
E C   I H   N
T     S     G

解析题目：

      使用多行存储，在将每一行拼接起来。按照下面的方式进行count遍历存储

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <stack>
#include <queue>
#include <list>
#include <unordered_map>
#include <cstring>
#include <map>
#include <stdexcept>

using namespace std;

class Solution {
public:
	string convert(string s, int numRows) {
		if (s.size() < 2 || numRows <= 1)
			return s;

		bool flag = false;
		int count = 1;
		vector<string> str;
		str.resize(numRows);
		str[0].push_back(s[0]);
		for (int i = 1; i < s.size(); i++)
		{
			if (!flag)
			{
				str[count] += s[i];
				if (count+1 == numRows)
				{
					flag = true;
				}
				else
				{
					++count;
				}
			}
			else{
				if (count != 0)
					--count;
				str[count] += s[i];
				if (count == 0)
				{
					++count;
					flag = false;
				}
			}
		}
		string res;
		for (int i = 0; i < numRows; i++)
		{
			int my_col = str[i].size();
			for (int j = 0; j < my_col; j++)
			{
				res += str[i][j];
			}
		}
		return res;
	}
};


int main(int argc, char *argv[])
{
	Solution solution;
	string str = "ABCD";
	cout << solution.convert(str,2) << endl;

	system("pause");
	return 0;
}