【leetcode】6.ZigZag Conversion（c语言）

Description：

The string “PAYPALISHIRING” is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)
And then read line by line: “PAHNAPLSIIGYIR”
Write the code that will take a string and make this conversion given a number of rows:
string convert(string s, int numRows);

Example1:

Input: s = “PAYPALISHIRING”, numRows = 3
Output: “PAHNAPLSIIGYIR”

Example2:

Input: s = “PAYPALISHIRING”, numRows = 4
Output: “PINALSIGYAHRPI”

[ 题目链接 ]

边界检查：
输入"A" numRows=1

提交代码：

char* convert(char* s, int numRows) {
   int i, k,len=0;
	int times = 2*(numRows - 1);

    //边界
	if (numRows == 1)	return s;
    
	//计算字符串长度
	for (i = 0; s[i] != '\0'; i++)
		len++;

	//创建返回数组
	char* sr = (char*)malloc((len+1)*sizeof(char));

	//返回数组填值
	int j = 0;
	for (i = 0; i < numRows; i++)
	{
		for (k = 0; i + k * times < len; k++)
		{
			sr[j++] = s[i + k * times];
			//printf("%c", s[i+k*times]);
			if (i != 0 && i != (numRows - 1) && ((k + 1)*times - i) < len)
				sr[j++] = s[(k + 1)*times - i];
				//printf("%c", s[(k+1)*times - i]);
		}
	}
	sr[j] = '\0';
	//printf("%s\n", sr);

	return sr; 
}

运行结果：