leetcode 6:Z字形变换
观察本题可以找到规律,每次对s的2*numRows-2个字母进行操作,比如3行时,每四个字母为一个循环,4行时每6个字母为一个循环。对于每次循环的前numRows个字母,用该字母的下标i%2*numRows 结果为当前字母所在的行。而对于每次循环的后numRows-2个字母,numRows-i%2*numRows为当前字母所在的行。
以numRows举例,每6个字母为一个循环,就拿前6个字母举例,前四个字母分别在0,1,2,3行,而对于第5个字母,下标为4,6-4%6=2,所以在下标为2的行,第6个字母,下标为5,同理,在下标为1的行
std::string convert(std::string s, int numRows) {
if(numRows<=0) return NULL;
if(numRows==1) return s;
std::vector<std::vector<char>> a;
int len=2*numRows-2;
for(int i=0;i<numRows;i++){
std::vector<char> v;
a.push_back(v);
}
for(int i=0;i<s.size();i++){
for(int j=0;j<numRows;j++){
if(i%len==j){
a[j].push_back(s[i]);
}
}
for(int j=numRows;j<len;j++){
if(i%len==j){
a[len-j].push_back(s[i]);
}
}
}
char c[s.size()];
int i=0;
for(int j=0;j<a.size();j++){
for(int k=0;k<a[j].size();k++){
c[i++]=a[j][k];
}
}
std::string str=c;//char数组转为字符串
return str.substr(0,s.size());//char转字符串会多出一个‘\0’,所以要截取
}