题目
将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "PAYPALISHIRING" 行数为 3 时,排列如下:
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
示例 3:
输入:s = "A", numRows = 1
输出:"A"
提示:
1 <= s.length <= 1000
s 由英文字母(小写和大写)、',' 和 '.' 组成
1 <= numRows <= 1000
我的思路是先进行分组,然后每一组根据索引对数据进行取出,最后按照新的排列方式返回即可
class Solution {
public:
string convert(string s, int numRows) {
int s_len = s.length();
string result_s;
char *ch_s = new char[s_len+1];
char *res_s = new char[s_len+1];
strcpy(ch_s,s.c_str());
if(numRows >= s_len || numRows == 1)
{
return s;
}
//分组
int group_num = 0;
group_num = numRows + numRows - 2;
//cout<<group_num<<endl;
int group_mod = 0;
int char_mod = 0;
int char_num_flag[s_len];
for(int i = 0;i<s_len;i++)
{
cout<<ch_s[i]<<",";
group_mod = i%group_num;
//char_mod = i%(group_num-2);
cout << group_mod <<endl;
if(group_mod < numRows)
{
char_num_flag[i] = group_mod;
}
else
{
char_num_flag[i] = numRows - 1 - (group_mod%numRows + 1);
}
}
int index = 0;
for(int i = 0;i< numRows;i++)
{
for(int j = 0; j < s_len ;j++)
{
//cout<<"-----"<<char_num_flag[j]<<endl;
if(char_num_flag[j] == i)
{
res_s[index] = ch_s[j];
index++;
}
if(index > s_len)
{
break;
}
}
if(index > s_len)
{
break;
}
}
res_s[s_len] ='\0';
result_s.assign(res_s);
cout << result_s <<endl;
return result_s;
}
};