Z 字形变换(C语言)

题目：将一个给定字符串s根据给定的行数numRows，以从上往下、从左到右进行 Z 字形排列。

示例：

输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"
解释：

---

 以下代码将原字符串的每个字符按照Z字形规律放进二维数组，再将数组按行取出，得到变换后的字符串。

char * convert(char * s, int numRows){
	int i, j = 0, k = 0;
    const int row = numRows;
    if(row==1||strlen(s)==1) return s;
    const int col = strlen(s)/(row-1) +1;
    char z[row][col+1];
    for(i=0;i<row;i++){
    	for(j=0;j<col+1;j++){
    		z[i][j] = '\0';
		}
	}
	
    j=0;
    for(i=0;i<strlen(s);i++){
        if(k%2==0){
            z[j][k] = s[i];
            j++;
            if(j==numRows-1){
                z[j][k] = '\0';
                k++;
            }
        }
        else{
            if(j==0){
                z[j][k] = '\0';
                k++;
                i--;
                continue;
            }
            z[j][k] = s[i];
            j--;
        }
    }
    j=0;k=0;
    for(i=0;i<col;i++){
    	if(z[0][i]!=0){
    		z[0][j] = z[0][i];
    		z[0][j+1] = '\0';
    		j++;
    	}
    	if(z[numRows-1][i]!=0){
    		z[numRows-1][k] = z[numRows-1][i];
    		z[numRows-1][k+1] = '\0';
    		k++;
		}
	}
	z[0][j] = '\0';
	z[numRows-1][k] = '\0';
//	puts(z[1]);
//	printf("%c\n", z[2][3]);
	k=0;
	for(i=0;i<row;i++){
		for(j=0;j<col+1;j++){
			if(z[i][j]==0){
				break;
			
			}
			
			s[k] = z[i][j];
			k++;
		}
	}
	return s;
    
}