ジグザグ変換（C言語）

トピック: 指定された行数 numRows に従って、指定された文字列 s を上から下、左から右にジグザグに配置します。

例：

入力: s = "PAYPALISHIRING"、numRows = 4
出力: "PINALSIGYAHRPI
"

---

 次のコードは、元の文字列の各文字をジグザグ規則に従って 2 次元配列に配置し、その配列を行ごとに取り出して、変換された文字列を取得します。

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;
    
}