题目来源:https://leetcode.cn/problems/zigzag-conversion/description/
C++题解:按行读取,每数列的索引间隔是相同的,除了首行和尾行,还有一个中间元素,循环用数组的长度来控制。
class Solution {
public:
string convert(string s, int numRows) {
int len = s.size();
int jj = 0, jjj = 0; //jj为索引,jjj为第几行
int aa = 2*numRows - 2, mm = 0; //aa是列的间隔,为固定值;mm为中间行的中间列索引间隔
string daan = "";
if((numRows >= len)or(numRows == 1)) return s; //特殊情况
daan = daan + s[0]; //首个元素
for(int ii = 0; ii < len; ii++){ //用len控制循环次数
if((jjj == 0)or(jjj == numRows-1)){ //第一行和最后一行,每次加一个
if(jj + aa < len){
jj = jj + aa;
daan = daan + s[jj];
}
else { // 跳出范围换行
jjj++;
jj = jjj;
if(jjj != numRows) daan = daan + s[jj];
}
}
else{ //中间行,每次要加两个
mm = aa - 2*jjj;
if(jj + mm < len) { // 判断斜线的元素
daan = daan + s[jj+mm];
if(jj + aa < len){ // 判断下一列,aa是固定的间隔
jj = jj + aa;
daan = daan + s[jj];
ii++; //一次加两个元素,新字符索引也要增长
}
else { // 跳出范围换行,再加上新行的第一个元素
jjj++;
jj = jjj;
daan = daan + s[jj];
ii++; //一次加两个元素,新字符索引也要增长
}
}
else { // 跳出范围换行,再加上新行的第一个元素
jjj++;
jj = jjj;
daan = daan + s[jj];
}
}
}
return daan;
}
};