【LeetCode 6】Z 字形变换

题目描述

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。（出题的人指定有点问题奥，明明是 И 字形变换，非要说成Z）
比如输入字符串为 “PAYPALISHIRING” 行数为 3 时，排列如下：

P   A   H   N
A P L S I I G
Y   I   R

之后，输出需要从左往右逐行读取，产生出一个新的字符串，比如：“PAHNAPLSIIGYIR”。
请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例 1

输入：s = "PAYPALISHIRING", numRows = 3
输出："PAHNAPLSIIGYIR"

示例 2

输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"
解释：
P     I    N
A   L S  I G
Y A   H R
P     I

示例 3

输入：s = "A", numRows = 1
输出："A"

思路

1       9        17
2     8 10     16 .
3   7   11   15   .
4 6     12 14     .
5       13        . . 

其实就是按照上述方式对字符串S进行排列，然后一行一行的拼接起来成为一个新的字符串并返回。

string convert(string s, int numRows) {
	string ans;
	if(numRows==1)
		return s;
	vector<string> a(numRows);
	int i=0;
	int flag=0;//用于判读目前的行的遍历顺序  0：由上到下  1：由下往上
	for(int  k=0;k<s.length();k++){
		if( flag==0 && i<numRows-1 ){//由上往下 没有到达边界
			a[i].push_back(s[k]);
			i++;
		}
		else if(flag==0 && i==numRows-1){//由上往下 到达边界
			a[i].push_back(s[k]);
			flag=1;
			i--;
		}
		else if( flag==1 && i>0 ){//由下往上 没有到达边界
			a[i].push_back(s[k]);
			i--;
		}
		else if(flag==1 && i==0){//由下往上 到达边界
			a[i].push_back(s[k]);
			flag=0;
			i++;
		}
	}
	
	for(int ii=0;ii<numRows;ii++)
		ans+=a[ii];
		
	return ans;
}

题目不难，主要是思路，也考验数据的结构操作。