题目要求:
The string "PAYPALISHIRING"
is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)
P A H N A P L S I I G Y I R
And then read line by line: "PAHNAPLSIIGYIR"
Write the code that will take a string and make this conversion given a number of rows:
string convert(string text, int nRows);
convert("PAYPALISHIRING", 3)
should return "PAHNAPLSIIGYIR"
.
所谓的ZigZag,就是这样的:(P.S:某些浏览器的高速模式可能看不到图,兼容或IE模式应该可以)
因为这是有规律的,可以用数学公式计算出下一个数组下标,这样节省了一定的空间和时间。 题目的输出是按行来拼接字符的,
因此我们可以通过逐行计算下标,拼接字符来达到目的。
通过画图观察,我们可以发现,
第0行和最后一行中,前一个下标的值和后一个下标的值相差 2 * nRows - 2 ,以第0行为例,前一个下标为0,
后一个下标为 0+2*4-2=6。
中间行中,前一个下标的值和后一个下标的值需要根据这个下标是该行中的奇数列还是偶数列来计算。以平时的习惯
来计算,因此,行和列的开始值都是0。
以第2行为例,第一个下标是2,后一个下标所处列为1,是奇数列,因此从这个下标到下一个下标相差的值是它们所处
的行i下面的所有行的点的个数,即2 * (nRows - 1 - i)。在这里,nRows-1是为了遵循0下标开始的原则。这样,我们可以
求得这个奇数列的下标为 2+2*(4-1-2)=4。同理,当我们要计算4之后的下标时,我们发现下一个所处的是偶数列2,从这个
下标到下一个下标相差的值其实是它们所处的行i上面的所有行的点的个数,即2 * i。因此,该列的下标计算为4+2*2=8。
有了以上的公式,我们就可以写出代码了。但需要注意几点:
1.判断所求出的下标是不是越界了,即不能超出原字串的长度;
2.注意处理nRows=1的情况,因为它会使差值2 * nRows - 2的值为0,这样下标值会一直不变,进入死循环,产生
Memory Limit Exceeded 或者 Output Limit Exceeded之类的错误。
3.还有一些其他的边界情况,如:nRows > s.length,s.length = 0等需要处理。
注意了以上事项后,就可以安全通过了。
以下是我的代码,欢迎各位大牛指导交流~
AC,Runtime: 80 ms
//LeetCode_ZigZag Conversion //Written by zhou //2013.11.4 class Solution { public: string convert(string s, int nRows) { // IMPORTANT: Please reset any member data you declared, as // the same Solution instance will be reused for each test case. if (nRows <= 1 || s.length() == 0) return s; string res = ""; int len = s.length(); for (int i = 0; i < len && i < nRows; ++i) { int indx = i; res += s[indx]; for (int k = 1; indx < len; ++k) { //第一行或最后一行,使用公式1: if (i == 0 || i == nRows - 1) { indx += 2 * nRows - 2; } //中间行,判断奇偶,使用公式2或3 else { if (k & 0x1) //奇数位 indx += 2 * (nRows - 1 - i); else indx += 2 * i; } //判断indx合法性 if (indx < len) { res += s[indx]; } } } return res; } };