将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 "LEETCODEISHIRING" 行数为 3 时,排列如下:
L C I R
E T O E S I I G
E D H N
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:"LCIRETOESIIGEDHN"。
请你实现这个将字符串进行指定行数变换的函数:
string convert(string s, int numRows);
示例 1:
输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"
示例 2:
输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:
L D R
E O E I I
E C I H N
T S G
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zigzag-conversion
关键点:可以使用分类,最顶层与最底层分开处理,中间部分可以分开处理。其中,发现规律,排列是循环的,可以看到,每2*Rows-2循环一次,利用这个特点进行处理,代码如下:
#include <stdio.h>
#include <string.h>
char * convert(char * s, int numRows){
int numA = 0, numB=0;
int row=0;
int Znum = 2*numRows - 2;
char *pchar = NULL;
int Array = strlen(s)/Znum + 1;
int diss = 0;
int lenth=strlen(s);
if(numRows <= 1)
return s;
pchar = (char *)malloc(lenth+2);
for(int i=0; i<numRows; i++)
{
diss = Znum-i;
for(int j=0; j<Array; j++)
{
if(i==0 || i==numRows-1)
{
if((i+j*Znum)>=lenth)
continue;
pchar[numA] = s[j*Znum+i];
numA++;
}
else
{
if((i+j*Znum)>=lenth)
continue;
pchar[numA] = s[i+j*Znum];
numA++;
if((j*Znum+diss)<lenth)
{
pchar[numA] = s[j*Znum+diss];
numA++;
}
}
}
}
return pchar;
}
void main()
{
char a[1000];
int numRow = 0;
char *ptest = NULL;
scanf("%s",a);
scanf("%d",&numRow);
ptest=convert(a, numRow);
printf("%s\n", ptest);
free(ptest);
return;
}
代码本身没有优化,自测OK,请各位指教。