概述
将一个给定字符串 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
思路:
我最先想到的就两种方法就是直接按照题目的描述,先将字符串存存储到二维数组中,然后再按行提取出来;二是找规律,但是我试了下老有问题所以我选择使用第一种笨方法。
//1.将字符串存储到一个行数为numRows的二维数组,浪费空间且麻烦
//2.直接找到规律
char arr[1000][1000]
= {0};
char * convert(char * s, int numRows){
if(numRows == 1)//一行的时候直接输出
return s;
for(int i = 0; i < numRows; i++) //先清空需要的行数,方便下面提取字符时判断
memset(arr+i, 0, 1000);
int len =
strlen(s);
int flag = 1, k = 0;
//flag用来标记是向下走,还是向斜上走
int x = 0, y = 0;
for(int i = 0; i < len; i++){
if(flag){
arr[x++][y] = s[i];
if(x == numRows - 1)
//走到行数最大值时要向斜上走
flag = 0;
}else{
arr[x--][y++] = s[i];
if(x == 0)
//行数最到0后要向下走
flag = 1;
}
}
char *ans = malloc(1020);
for(int i = 0; i < numRows; i++){
//按行遍历
for(int j = 0; j <= y; j++){
if(arr[i][j])
//如果存储的有字符,则就是我们需要的字符
ans[k++] = arr[i][j];
}
}
ans[len ] = '