这道题是LeetCode里的第6道题。
题目要求:
将一个给定字符串根据给定的行数,以从上往下、从左到右进行 Z 字形排列。
比如输入字符串为
"LEETCODEISHIRING"
行数为 3 时,排列如下:L C I R E T O E S I I G E D H N
之后,你的输出需要从左往右逐行读取,产生出一个新的字符串,比如:
"LCIRETOESIIGEDHN"
。请你实现这个将字符串进行指定行数变换的函数:
string convert(string s, int numRows);
示例 1:
输入: s = "LEETCODEISHIRING", numRows = 3 输出: "LCIRETOESIIGEDHN"
示例 2:
输入: s = "LEETCODEISHIRING", numRows = 4 输出: "LDREOEIIECIHNTSG" 解释: L D R E O E I I E C I H N T S G
题目虽然很长,但是题目本身并不难,却相当的花时间。这道题l类似于我之前做过的螺旋矩阵(回形取数),打打草稿发现规律,找出函数关系就行了。
拿例子2说明:
输入:s = “LEETCODEISHIRING”,numRows = 4
Z字排列如下:
转换成对应的下标索引为:
首先可以发现第一行的索引添加量和Z字的行数numRows有关系:IndexAdd=2*numRows-2
例如当 numRows = 3 时:
IndexAdd=2*3-2=4
根据这个特性首先可以解决第一行,同理解决最后一行。
解决完首尾行后中间行怎么办呢?因为中间行有斜对角的字符,要如何处理?其实不难,拿例子2来说,第二行1和7中间一个,7和13中间一个,第三行2和8中间一个,8和14中间一个,也就是说中间行每隔 IndexAdd 就会有一个字符,前提是不超出范围。numRows = 3 时,中间第二行也符合这个规律。而且这些字符还符合另一规律:例子2中5在 i=1 行(索引值),1和5差4,4在 i=2行(索引值),2和4差2。也就是说中间字符和左边字符的差值为 IndexAdd-2*i,所以中间行这样处理:第一个是 j=i,然后中间字符 j=j+IndexAdd-2*i,再是 j=j+2*i,这样走完一轮相当于 j=j+IndexAdd,一直循环。
说明:i 是行的索引值,不是第 i 行。第一行的索引值为0,第二行的索引值为1,等等。
提交代码:
class Solution {
public:
string convert(string s, int numRows) {
int i=0,j,len=s.length(),IndexAdd=(numRows-1)*2;
if(numRows==1||numRows>=s.length())return s;
string res;
while(1){
j=i;
if(i==0||i==numRows-1){
while(j<len){res+=s[j];j=j+IndexAdd;}
i++;
if(i==numRows)break;
}else{
while(j<len){res+=s[j];j=j+IndexAdd-2*i;
if(j<len){res+=s[j];j=j+2*i;}}
i++;
}
}
return res;
}
};
提交结果:
个人总结:
题目不难只要发现其中规律就能很好的解决。一开始处理的时候弄混淆了下标和字符的关系,浪费了些时间。