题目
将一个给定字符串 s
根据给定的行数 numRows
,以从上往下、从左到右进行 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"
C++代码
#include <iostream>
#include <vector>
using namespace std;
/*
* 定义N字形变换函数
* 参照案例,我们可以发现在行索引为0和numRows-1的位置为临界点,
* 在这里可以定义一个初始值为-1的flag,如果行索引为临界点,反向执行
* 定义新的字符串数组,将字符放入,更新索引
* 遍历数组,输出变换后的字符串
*/
string convert(string s, int numRows) {
if (numRows < 2) {
return s;
}
vector<string> rows(numRows);
int i = 0, flag = -1;
for (char c : s) {
rows[i].push_back(c);
if (i == 0 || i == numRows - 1) {
flag = -flag;
}
i += flag;
}
string res;
for (const string& row : rows) {
res += row;
}
return res;
}
int main() {
string s = "PAYPALISHIRING";
int numRows = 3;
string res = convert(s, numRows);
cout << "The converted s: " << res << endl;
return 0;
}
分析
定义N字形变换函数 convert(),参照案例,我们可以发现在行索引为 0 和 numRows - 1 的位置为临界点,在这里可以定义一个初始值为 -1 的 flag,如果行索引为临界点,反向执行。定义新的字符串数组 rows,将字符 c 放入,更新索引 i ,遍历数组,输出变换后的字符串 res。
问题
将 Z 字形看作 N 字形变形,观察行索引上的元素索引变化,可以发现规律,再运用 flag 来更新索引,排列字符。在排列时已经把每行的字符都放入了字符串数组,可以输出。
可以直接用 for(char c : s) 遍历字符串 s 的元素。