6. ZigZag Conversion LeetCode题解

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)

字符串

"PAYPALISHIRING"

 写成左旋右旋的形式如下所示：

P   A   H   N
A P L S I I G
Y   I   R

And then read line by line: 

"PAHNAPLSIIGYIR"

按行读取，则是 

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

.

Subscribe to see which companies asked this question.

题解：
字符串类题目，考虑两种思路：
1. 按照题意方式进行模拟，对于每一行开一个StringBuilder，按照旋转顺序，依次将字符插入到对应行的StringBuilder中，最后将这些StringBuilder拼起来；
2. 直接计算新字符串中下一个字符在原字符中的位置；
代码采取第二种思路，不难发现，zig旋转方向的下一个字符计算公式为k * (n -1 + n - 2 + 1) + line = k * (n * 2 - 2) + line
其中k表示第几列zig，n表示共有几行，line表示当前行数
同样，如果改行存在zag，则公式为(k + 1) * (2 * n - 2) - line

Code【Java】
public class Solution {
private boolean addChar(StringBuilder sb, String s, int pos) {
if (pos < 0 || pos >= s.length()) {
return false;
}
sb.append(s.charAt(pos));
return true;
}
public String convert(String s, int numRows) {
// 处理特殊情况
if (s == null || s.length() <= numRows || numRows < 2) {
return s;
}
StringBuilder sb = new StringBuilder();
for (int i = 0; i < numRows; ++i) {
for (int k = 0; ; ++k) {
// 竖直方向添加字符
int pos1 = k * (2 * numRows - 2) + i;
if (!addChar(sb, s, pos1)) break;
// 倾斜方向添加字符
if (i == 0 || i == numRows - 1) {
continue;
}
int pos2 = (k + 1) * (2 * numRows - 2) - i;
if (!addChar(sb, s, pos2)) break;
}
}
return sb.toString();
}
}

Code【C++】
class Solution {
bool addChar(stringstream& ss, string& s, int pos) {
if (pos < 0 || pos >= s.length()) {
return false;
}
ss << s[pos];
return true;
}
public:
string convert(string s, int numRows) {
// 处理特殊情况
if (s.length() <= numRows || numRows < 2) {
return s;
}
stringstream ss;
for (int i = 0; i < numRows; ++i) {
for (int k = 0; ; ++k) {
// 竖直方向添加字符
int pos1 = k * (2 * numRows - 2) + i;
if (!addChar(ss, s, pos1)) break;
// 倾斜方向添加字符
if (i == 0 || i == numRows - 1) {
continue;
}
int pos2 = (k + 1) * (2 * numRows - 2) - i;
if (!addChar(ss, s, pos2)) break;
}
}
return ss.str();
}
};