leetcode-6. Z 字形变换

leetcode-6. Z 字形变换。

将一个给定字符串根据给定的行数，以从上往下、从左到右进行Z字形排列。
请实现这个将字符串进行指定行数变换的函数：char *convert( char *s, int numRows );

示例 1:

输入: s = "LEETCODEISHIRING", numRows = 3
输出: "LCIRETOESIIGEDHN"
解释:
L   C   I   R
E T O E S I I G
E   D   H   N

示例 2:

输入: s = "LEETCODEISHIRING", numRows = 4
输出: "LDREOEIIECIHNTSG"
解释:
L     D     R
E   O E   I I
E C   I H   N
T     S     G

示例 3:

输入: s = "PAYPALISHIRING", numRows = 4
输出: "PINALSIGYAHRPI"
解释:
P    I    N
A  L S  I G
Y A  H R
P    I

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zigzag-conversion/
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一：辅助矩阵 + 暴力循环。

char *convert( char *s, int numRows ) {
	char *tempA = NULL;
	int i = 0, r = 0, c = 0, up = 0;
	int col = strlen( s );

	if( numRows < 2 || numRows == col ) {
		return s;
	}
	tempA = calloc( sizeof(*tempA), numRows * col );
	for( i = 0; s[i] != ''; ++i ) {
		tempA[r * col + c] = s[i];
		if( !up ) {
			if( r == numRows - 1 ) {
				// 从上往下走,到最后一个位置时.
				r = numRows - 2;
				++c;
				up = 1;
			} else {
				++r;
			}
		} else {
			if( r == 0 ) {
				// 从下往上走,到最后一个位置.
				r = 1;
				up = 0;
			} else {
				--r;
				++c;
			}
		}
	}
	for( i = r = 0; r < numRows; ++r ) {
		for( c = 0; c < col; ++c ) {
			if( tempA[r * col + c] != '' ) {
				s[i++] = tempA[r * col + c];
			}
		}
	}
	free( tempA );

	return s;
}

解法二：模拟动态字符串。

用上面的示例3进行解释: 输入: s = "PAYPALISHIRING", numRows = 4, 输入: "PINALSIGYAHRPI".

根据示例3明显得到: P、I、N 位于第1(下标0)行,
                   A、L、S 、I、G 位于第2(下标1)行,
                   Y、A、H、R 位于第3(下标2)行,
                   P、I 位于第4(下标3)行.

可以把每一行看做为独立的容器，
当遍历计算得到每个字符所位于的行，就将字符加入到对应的容器中，例如：将 P、I、N 加入到容器1中,将 P、I 加入到容器4中.
遍历结束时得到: 容器1包含P、I、N, 容器1包含A、L、S、I、G, 容器1包含Y、A、H、R, 容器1包含P、I.
把每个容器依次拼接得到 "PINALSIGYAHRPI", 即是最终结果.

参考: https://leetcode-cn.com/problems/zigzag-conversion/solution/zzi-xing-bian-huan-by-jyd/

char *convert( char *s, int numRows ) {
	char *tempA = NULL;
	int *column = NULL;
	int i = 0, r = 0, c = 0, up = 0, col = strlen( s );

	if( numRows < 2 || numRows == col ) {
		return s;
	}
	tempA = malloc( sizeof(*tempA) * numRows * col );
	column = calloc( sizeof(*column), numRows );
	for( i = 0; s[i] != ''; ++i ) {
		tempA[r * col + column[r]++] = s[i];
		if( !up && ++r == numRows ) {
			r = numRows - 2;
			up = 1;
		} else if( up && --r < 0 ) {
			r = 1;
			up = 0;
		}
	}
	for( r = 0; r < numRows; ++r ) {
		tempA[r * col + column[r]] = '';
	}
	free( column );
	s[0] = '';
	for( r = 0; r < numRows; ++r ) {
		strcat( s, &tempA[r * col] );
	}
	free( tempA );

	return s;
}