LeetCode6 Z字形变换

题目

将字符串 “PAYPALISHIRING” 以Z字形排列成给定的行数：
P A H N
A P L S I I G
Y I R
之后从左往右，逐行读取字符：“PAHNAPLSIIGYIR”
实现一个将字符串进行指定行数变换的函数:
string convert(string s, int numRows);

示例

1. 示例 1:
   输入: s = “PAYPALISHIRING”, numRows = 3
   输出: “PAHNAPLSIIGYIR”
2. 示例 2:
   输入: s = “PAYPALISHIRING”, numRows = 4
   输出: “PINALSIGYAHRPI”
   解释:
   P I N
   A L S I G
   Y A H R
   P I

C++代码

首先进行分析。对应下标的关系。
将第一行和最后一行单独列出（因为单个z字里边，第一行和最后一行只有两个字母）。
对于中间的每一行，排除第一列，可以将其后每一个z字中的两列看为一个整体，两个字母之间的下标之差为2*i（i为第i行）。
注意到，因为退出循环的条件是下标超过了字符串长度，而代码中是以每一个z字最后一列为基准来进行读取的，所以有可能会出现最后一列的下标超出了字符串长度，但是其前面对应的字母的下标没有超出，所以最后还应该对前边字母的下标和字符串长度进行对比，以衡量是否读取该字母。

注意当行数为0或1，s为空时的特殊情况。

class Solution {
public:
	string convert(string s, int numRows){
		string r = "";
		if(s.length()==0||numRows==0||numRows==1) return s;
		int d = (numRows - 1) * 2;
		for (int i = 0; i<numRows; i++)
		{
			if (i == 0 || i == numRows - 1)
			{
				int j = i;
				while (j < s.length())
				{
					r += s[j];
					j += d;
				}
			}
			else
			{
				r += s[i];
				int j = i+d;
				while (j < s.length())
				{
					r = r + s[j - 2*i] + s[j];
					j += d;
				}
				if (j - 2 * i < s.length()) r += s[j - 2 * i];
			}
		}
     return r;
	}
}；