思想:
关键是归纳出元素出现位置的公式。
比如 nRows =4 的情形需要下面这样的分布:
第0行:0 + + 6 + + 12
第1行:1 + 5 7 + 11
第2行:2 4 + 8 10
第3行:3 + + 9
(1)对于 第0行 && 第3行(最后一行):
出现的元素在原串的位置依次是:0,6,12
规律是:0,0+2*nRows-2=6,6+2*nRows-2=12
即后面的每一个元素的位置都是前面元素位置加上(2*nRows-2)。
(2)对于其余的行,即中间的所有行:
比如第1行(设行标为i,此时i=2)出现的元素在原串的位置依次是:1,5,7,11
规律是:1,1+2*(nRows-1-i)=5,5+2*i=7,7+2*(nRows-1-i)=11
即后面的元素所在列+1如果是奇数,则下个元素位置则为该位置+2*(nRows-1-i);如果是偶数,则下个元素位置为该位置+2*i。
当然,往最后的结果串中加元素的话还要看下标是否越界。
class Solution {
public:
string convert(string s, int nRows) {
if(nRows <= 1 || s.length() == 0) {
return s;
}
string res = "";
int len = s.length();
for(int i=0; i<len && i<nRows; i++) {
int idx = i;
res += s[idx];
//看对每一行是否有可以加入到res上的元素
for(int k=1; idx < len; k++) {
//如果是 第0行 或者是 最后一行 的判断方法
if(i == 0 || i == nRows-1) {
idx += 2*nRows - 2;
//如果是 中间行 的判断方法
}else {
if(k & 0x1) {
idx += 2*(nRows-1-i);
}else {
idx += 2*i;
}
}
//看原串中是否存在该元素,存在就加入
if(idx < len) {
res += s[idx];
}
}
}
return res;
}
};