Leetcode: ZigZag Conversion

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".

一道数学题，参考了网上的思路，先计算一下每一个zig包含的字符个数，实际上是zigSize = 2*nRows – 2
然后一行一行的加s中的特定元素就行。
第一行和最后一行都只需要加一个字符，每一个zig，而且在s中的index间隔是zigSize。
中间每一行需要添加两个字符到结果中去。第一个字符同上；第二个字符的index就是zigSize – i。i是行index。

 需要注意的是，nRows == 1一定要单独讨论，否则rsize = 2*nRows - 2 == 0，循环没有增量会成为死循环。还有18-23行添加的顺序谁先谁后是有讲究的。

第三遍代码：361ms

 1 public class Solution {

 2     public String convert(String s, int nRows) {

 3         if (s == null || s.length() == 0) return "";

 4         if (s.length() <= nRows || nRows == 1) return s;

 5         int rsize = 2 * nRows - 2;

 6         StringBuffer res = new StringBuffer();

 7         for (int i=0; i<nRows; i++) {

 8             if (i == 0 || i == nRows-1) {

 9                 int t = 0;

10                 while (i + t * rsize < s.length()) {

11                     res.append(s.charAt(i+t*rsize));

12                     t++;

13                 }

14             }

15             else {

16                 int m = 0;

17                 while (i + m * rsize < s.length() || (rsize - i + m * rsize) < s.length()) {

18                     if (i + m * rsize < s.length()) {

19                         res.append(s.charAt(i+m*rsize));

20                     }

21                     if ((rsize - i + m * rsize) < s.length()) {

22                         res.append(s.charAt(rsize-i+m*rsize));

23                     }

24                     m++;

25                 }

26             }

27         }

28         return res.toString();

29     }

30 }

 如果不分开i =0, i=nRows-1和其他case, 444ms, 比上面反而慢一点

 1 public class Solution {

 2     public String convert(String s, int nRows) {

 3         if (s == null || s.length() == 0) return "";

 4         if (s.length() <= nRows || nRows == 1) return s;

 5         int rsize = 2 * nRows - 2;

 6         StringBuffer res = new StringBuffer();

 7         for (int i=0; i<nRows; i++) {

 8                 int m = 0;

 9                 while (i + m * rsize < s.length() || (rsize - i + m * rsize) < s.length()) {

10                     if (i + m * rsize < s.length()) {

11                         res.append(s.charAt(i+m*rsize));

12                     }

13                     if (i!=0 && i!=nRows-1 && ((rsize - i + m * rsize) < s.length())) {

14                         res.append(s.charAt(rsize-i+m*rsize));

15                     }

16                     m++;

17                 }

18 

19         }

20         return res.toString();

21     }

22 }