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’,所以要截取
}

猜你喜欢

转载自blog.csdn.net/u013263891/article/details/83241923