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"
.
题意:将一个字符串按给定行数蜿蜒排列,返回排列后按行输出的字符串。
我觉得这题的关键点在于弄清楚zigzag pattern的含义,下面我画图先解释一下:
(设输入字符串s="abcdefghijklmn……")
1、若nRows=1,则排列格式为abcdefghijklmn...;
2、若nRows=2,则排列格式为
3、若nRows=3,则排列格式为
4、若nRows=4,则排列格式为
5、若nRows=5,6,7....同理
由上我们可以观察到,这个排列是是有规律可循的,当nRows=1,2时,字符串以原本的顺序排列和插空排列。
nRows>2时,每个循环的元素个数为2(nRows-1)。因此,我们可以建立一个动态数组vector,里面包含nRows个
string类型的元素,然后对字符串s进行遍历,并将字符插入到对应的string中,最后将所有string相加,即可得到所要
的结果字符串。
class Solution {
public:
string convert(string s,int nRows)
{
if(s=="")
return s;
int len=s.size();
if(nRows==1)
return s;
else if(nRows==2)
{
string res="";
for(int i=0;i<len;i++,i++)
res+=s[i];
for(int i=1;i<len;i++,i++)
res+=s[i];
return res;
}
vector<string> res;
for(int i=0;i<nRows;i++)
res.push_back("");
for(int i=0;i<len;i++)
{
int j=i;
j=j%(2*nRows-2);
if(j<nRows)
res[j]+=s[i];
else
res[2*(nRows-1)-j]+=s[i];
}
string result="";
for(int i=0;i<nRows;i++)
result+=res[i];
return result;
}
};