LeetCode06——Z字形输出问题

题目：
将一个给定字符串根据给定的行数，以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为 “LEETCODEISHIRING” 行数为 3 时，排列如下：
L C I R
E T O E S I I G
E D H N
之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如：“LCIRETOESIIGEDHN”。请你实现这个将字符串进行指定行数变换的函数。

思想：
这题其实就是一个找规律的题目，观察输出结果的形式，除了第0行和最后一行，其余的行数字母的间隔数都有一定规律，找出这个规律直接遍历即可。

#include <stdlib.h>
#include <stdio.h>
#include <malloc.h>
#include <string.h>


char* convert(char* s, int numRows) {
    if(numRows<=1) return s;
    char *s1 = (char *)malloc(sizeof(char)*(strlen(s) + 1));
    int k = 0;
    int step = 2*numRows-2;
    for(int i=0; i<numRows; i++)
        for(int j=i; j<strlen(s); j+=step)
        {
            s1[k++] = s[j];
            int temp = j+step-2*i;
            if(i!=0 && i!=numRows-1 && temp<strlen(s))
            s1[k++] = s[temp];
        }
   s1[k]=0;
   return s1;
}

int main() {
    char str[] = "abcdefghij";
    printf("%s",convert(str,3));
    return 0;
}