题目描述
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时,排列如下:
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:“LCIRETOESIIGEDHN”。
请你实现这个将字符串进行指定行数变换的函数:
string convert(string s, int numRows);
示例 1:
输入: s = “LEETCODEISHIRING”, numRows = 3
输出: “LCIRETOESIIGEDHN”
示例 2:
输入: s = “LEETCODEISHIRING”, numRows = 4
输出: “LDREOEIIECIHNTSG”
解释:
题目解答:
方法1:找规律
该题是一个找规律的题,需要确定哪些位置的字符会放在同一行。通过分析可以看出来,对于某一行 i ,假定当前字符A位置为temp
,如果2n - 2i - 2
不为0,则同一行的下一个字符B位置就是temp + 2n - 2i - 2
,接下来如果2i
不为0,则下一个字符C位置为temp + 2i
(temp已更新为上一个字符B的位置),接下来继续重复上述过程,直到temp
大于字符串长度。运行时间12ms,代码如下。
char* convert(char* s, int numRows) {
int n = numRows, index = 0;
int len = strlen(s), temp = 0;
int i = 0, t = 2 * n - 2;
char* result = (char*)malloc((len + 1) * sizeof(char));
result[len] = '\0';
if(n == 1) {
memcpy(result, s, len * sizeof(char));
return result;
}
for(i = 0; i < n; i++) {
result[index++] = s[i];
temp = i;
while(1) {
if(t - 2 * i) {
if(temp + t - 2 * i < len) {
temp += (t - 2 * i);
result[index++] = s[temp];
}
else
break;
}
if(2 * i) {
if(temp + 2 * i < len) {
temp += 2 * i;
result[index++] = s[temp];
}
else
break;
}
}
}
return result;
}