题目:
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的图形如下图所示。
其实这一题,主要考察的是等差数列,纸上先推导好公式,这题就很简单的求解了~
class Solution {
public:
string convert(string s, int numRows) {
if(numRows==1||numRows>=s.size())
return s;
string res="";
for(int i=0;(numRows*2-2)*i<s.size();i++)
{
res+=s[(numRows*2-2)*i];
}
for(int i=1;i<numRows-1;i++)
{
int begin=1;
res+=s[i];
while(true)
{
if(-i+begin*(2*numRows-2)<s.size())
{
res+=s[-i+begin*(2*numRows-2)];
}
else break;
if((2*numRows-2)*begin+i<s.size())
{
res+=s[(2*numRows-2)*begin+i];
}
else break;
begin++;
}
}
for(int i=0;(numRows*2-2)*i+numRows-1<s.size();i++)
{
res+=s[(numRows*2-2)*i+numRows-1];
}
return res;
}
};