将字符串 "PAYPALISHIRING" 以Z字形排列成给定的行数:
P A H N
A P L S I I G
Y I R
之后从左往右,逐行读取字符:"PAHNAPLSIIGYIR"
实现一个将字符串进行指定行数变换的函数:
string convert(string s, int numRows);
示例 1:
输入: s = "PAYPALISHIRING", numRows = 3
输出: "PAHNAPLSIIGYIR"
示例 2:
输入: s = "PAYPALISHIRING", numRows = 4
输出: "PINALSIGYAHRPI"
解释:
P I N
A L S I G
Y A H R
P I
代码
public:
string convert(string s, int numRows)
{
if(numRows==1)//边界情况,如果只有1行,那么直接返回原本的字符串
return s;
int s1=s.size(),s2=2*numRows-2,s3=s2,index=0,index1,count=0;//s2表示山峰的元素个数
string res=s;
for(int i=0;i<numRows;i++)//逐行处理
{
index1=index;//记录一下index的值,最开始为0,接着是1,再接着是2……
while(index<s1)//每一行中只要坐标index不超过字符串的长度,那么不断处理
{
if(s3==s2)//如果是第一行
{
res[count]=s[index];
index+=s3;
count++;
}
else if(s3==0)//如果是最后一行
{
res[count]=s[index];
index+=(s2-s3);
count++;
}
else//如果是中间行
{
res[count]=s[index];
index+=s3;
count++;
if(index<s1)//不能超过字符串的长度
{
res[count]=s[index];
index+=(s2-s3);
count++;
}
}
}
s3-=2;//处理完一行之后,s3减去2
index=index1+1;//更新index的值
}
return res;//最后返回新的字符串
}