Leetcode : 6 ZigZag Conversion C++

描述

The string “PAYPALISHIRING” is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

P A H N
A P L S I I G
Y I R
And then read line by line: “PAHNAPLSIIGYIR”
Write the code that will take a string and make this conversion given a number of rows:

string convert(string text, int nRows);
convert(“PAYPALISHIRING”, 3) should return “PAHNAPLSIIGYIR”.

分析

就是将一个字符串“之”字型回填，然后按行输出。那不妨就用char型二维数组。这个麻烦就麻烦在只能确定行数，无法确定列数，所以不妨借助可以动态添加的vector来实现。只要还没有遍历到字符串最后，就按照自然的回填顺序，先从上往下填一列，再从下往上斜着填回去，最后迭代一些形成字符串即可

代码

string convert(string s, int numRows) {
        int len = s.length();
        vector<vector<char> >vec(numRows);
        int l=0;
        while(l<len){
        for(int i = 0;i<numRows&&l<len;i++)
        {
            vec[i].push_back(s[l]);l++;
        }
        for(int j =numRows-2;j>=1&&l<len;j--)
        {
            vec[j].push_back(s[l]);l++;
        }
        }
        vector<vector<char> >::iterator it1;
        vector<char>ve; vector<char>::iterator it2;
        string ans;
        for(it1=vec.begin();it1!=vec.end();it1++)
        {
            ve = *it1;
            for(it2=ve.begin();it2!=ve.end();it2++)
                //cout<<*it2;
                ans += *it2;
        }
        return ans;