审题
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 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
这道题看下来,看了几分钟后感觉是小模拟的感觉了,方法也很简单,用一个指针从第一行到最后一行再从最后一行到第一行,在其中,我们每次都把指针指到的元素 i i i附加到 v e c t o r < s t r i n g > v [ i ] vector<string> v[i] vector<string>v[i]中,最后再将每一个元素从0到 n − 1 n-1 n−1加到一个新字符串里就好啦。
代码实现
代码中需要注意一些的是特判,也就是当numRows为0的时候,要注意cnt%0的情况,直接返回原字符串就可以了。当然,我这里判断的条件是找的规律,其实完全没有必要这么麻烦…直接用0和numRows-1作为边界条件判断就可以了。
代码如下:
class Solution {
public:
string convert(string s, int numRows) {
int flag=1, cnt=0;
string fin;
vector<string> v(numRows, "");
if ( numRows == 1 ) return s;
for ( int i=0; i<s.size(); i++ ) {
if ( cnt % (2*numRows-2) == 0 && i != 0 ) flag = -flag;
else if ( cnt % ((2*numRows-2)-(numRows-1)) == 0 && i != 0 ) flag = -flag;
v[cnt] += s[i];
cnt += flag;
}
for ( int i=0; i<numRows; i++ ) {
fin += v[i];
}
return fin;
}
};
反思
这道小模拟题属于需要动一点点脑筋的类型,熟练掌握stl和字符串操作,解决应该不是很难。