c语言实现Z字形变换

题目描述：将一个给定字符串 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”

//Z字形变换
//思路：用二维数组存储；再来回存储；
//典型错误：只要存字符串，最后一个字符一定要置为’\0’

#include<stdio.h>
#include<string.h>
char * convert(char * s, int numRows){
	char a[numRows][60];
	int n=0,count=0,flag,t=0,c,i,j,k=0;
	for(i=0;i<numRows;i++)//5否则a[i][0]!=NULL 
	{
		a[i][0]='\0';
	}
	while(s[n]!='\0')//求数组长度
	{
		n++;//n最后的长度为数组长度加一 
	}
	char hh[n+20];
	for(i=0;i<n;i++)//核心代码：将字符串按z字形存入（来回存） 
	{
		c=strlen(a[count]);//获取数组长度，以便知道字符应该存在那个位置 
		a[count][c]=s[i];
		a[count][c+1]='\0';
		if(count==(numRows-1)) 
		{
			flag=1;//此时降序读,flag作为转择标记 
		}
		if(count==0)
		{
			flag=0;//此时升序读 
		} 
		if(flag==1)
		count--;
		if(flag==0)
		count++; 
	} 
	for(i=0;i<numRows;i++)//将n个数组存入一个数组中 
	{
		c=strlen(a[i]);//获取二维数组的长度
		for(j=0;j<c;j++)
		{
			hh[k]=a[i][j];
			k++;
		} 
	 } 
	 hh[k]='\0';
	 
	 return hh; 
}
int main(){
	int n;
	scanf("%d",&n);
	char ch,h[20];//注意：数组的初始值 
	int i=0;
	while((ch=getchar())!='\n')//读入字符串到数组h[] 
	{
		h[i]=ch;
		i++;
	} 
	h[i]='\0';
	char * p=convert(&h[0],n);//接收产生的新数组 
	for(int j=0;j<i;j++)
	{
		printf("%c",p[j]);
	}
	return 0;
}