力扣——Z字符串变换

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
string convert(string s, int numRows)
{
	if (numRows == 1)
	{
		return s;
	}
	vector<string> restr(min(numRows, int(s.size())));
	int j = 0;
	int flag = -1;
	for (char c:s)//复制一个字符串，再进行遍历操作
	{
		restr[j] += c;
		if (j == 0 || j==numRows-1)
		{
			flag = -flag;
		}
		j += flag;
	}
	string ret;
	for (string row :restr) ret += row;
	return ret;
}
int main()
{
	string s;
	int numRows;
	cout << "input a string: ";
	cin >> s;
	cout << "input a numRows: ";
	cin >> numRows;
	cout << "the convert string is: "<<convert(s, numRows) << endl;
	return 0;
}