题目描述:
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 RAnd 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"
.------------------------------------------------------------------------------------------------------------------------------------------------
上述例子不容易发现规律,特意查找了相关的帖子,举了nRows = 4的例子:
text = "ABCDEFGHIJKLMN", n = 4,转换后字符串为 "AGMBFHLNCEIKDJ"
A G M B F H L N C E I K D J在nRows = 3和nRows = 4的情况下,字母回到第一行共经历2*nRows-2次(即4次和6次)变换,记录n = 2 * nRows - 2;
遍历字符串s,计算i % n,当i % n < nRows时,直接将该字母存入s_array[i % n];当i % n > nRows时,寻找规律,将字母存入s_array[n - (i % n)];
最后将s_array临时数组进行首尾相接,得到最终结果。
-------------------------------------------------------------------------------------------------------------------------------------------------
代码如下:
class Solution {
public:
string convert(string s, int numRows) {
if(numRows == 1)
return s;
string* s_array = new string[numRows];
string result = "";
for(int i = 0; i < s.length(); i++)
{
int n = 2 * numRows - 2;
if(i%n < numRows)
s_array[i%n] += s[i];
else
s_array[n-(i%n)] += s[i];
}
for(int i = 0; i < numRows; i++)
{
result += s_array[i];
}
return result;
}
};