思路:首先计算出组数,对于nRows的矩阵,奇数列是垂直的,占nRows个元素,偶数列是斜对角的,占nRows -2 个元素,所以一组共有 2 *nRows - 2个元素,除非nRows= 1时还是1,不能根据这个公式计算,直接返回原字符串即可。
然后计算每个元素s[i]在矩阵中的行列位置,对于行,先用 i 对组数groupNum取模得到R,如果小于行数,则不变,如果大于行数则更正为 groupNum - R,对于列数,只要i除以groupNum即可,如果前面取模的R大于行号,说明在斜线部分,列数加个1,否则不变。
至此,行列就求出来了,剩下的根据矩阵按行存入字符串中即可。
code:
class Solution {
public:
string convert(string s, int nRows) {
if(s.empty())
return s;
int len = s.length();
int groupNum = 2*nRows-2;
if(nRows == 1)
return s;
int nCols = (len / groupNum + 1)* 2;
string str(nCols,' ');
vector<string> converStr(nRows,str);
string ret;
for(int i = 0; i < len;i++){
int r = i % groupNum;
int c = 2*(i / groupNum);
if(r >= nRows){
r = groupNum - r;
c += 1;
}
converStr[r][c] = s[i];
}
for(int i = 0;i < nRows;i++){
for(int j = 0;j < nCols;j++){
if(converStr[i][j] != ' ')
ret += converStr[i][j];
}
}
return ret;
}
};