将一个给定字符串 s 根据给定的行数 numRows ,以从上往下、从左到右进行Z字形排列。之后,你的输出需要从左往右逐行读取,产生出一个新的字符串
示例 : 输入: s ="QAZWSXEDCRFVTG",numRows = 4 输出:"QETAXDVGZSCFWR"
A->X->D对应代码中的注释部分
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
char* convert(char* s, int numRows) {
if (numRows == 1) {
return s; // 如果行数为1,直接返回原字符串
}
int len = strlen(s);
char* ans = (char*)malloc((len+1)*sizeof(char));
ans[len] = '\0'; // 新建字符串数组并且初始化为'\0'
int k = 0;
int step = (numRows-1)*2; // 每个Z字形的步长
for (int i = 0; i < numRows; i++) {
int index = i;
int diff = (numRows-i-1)*2;
/*******Begin********/
int diff2 = i*2;
while(index < len) {
if(i != 0 && i != numRows-1) { // 非首尾行需要考虑两种跨度
if(index < len) {
ans[k++] = s[index]; //A
}
index += diff;
if(index < len) {
ans[k++] = s[index]; //X
}
index += diff2; //D
} else { // 首尾行只需考虑一个跨度
if(index < len) {
ans[k++] = s[index];
}
index += step;
}
}
}
return ans;
}
int main() {
char s[20];
scanf("%s", &s);
int numRows;
scanf("%d", &numRows);
printf("%s", convert(s,numRows));
return 0;
}
结果