LeetCode: ZigZag Conversion

思路:首先计算出组数,对于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;
    }
};


你可能感兴趣的:(LeetCode: ZigZag Conversion)