码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode]ZigZag Conversion

时间:2015-01-26 13:27:13      阅读:148      评论:0      收藏:0      [点我收藏+]

标签:

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".

 

思路,用一个idx 从0-->(nRows-1)---->0 来回往返,将对应的字符加入到strs[idx]的字符串中,最后汇总所有的字符串,

 

注意,当nRows == 1的情况需要特使处理,否则idx-- 后,idx变成-1,会越界crash

class Solution {
    public:
        string convert(string s, int nRows) {
            if(nRows == 1)
                return s;
                
            vector<string> strs;
            strs.resize(nRows);
            string result;

            int idx = 0;
            bool upOrDown = false;// false--down, true--up

            for(size_t i = 0; i< s.size(); i++)
            {   
                strs[idx] += s[i];
                if(upOrDown == false)
                {   
                    if(idx == (nRows-1))
                    {   
                        upOrDown = true;
                        idx--;
                    }   
                    else
                    {   
                        idx++;
                    }   
                }   
                else
                {   
                    if(idx == 0)
                    {   
                        upOrDown = false;
                        idx++;
                    }   
                    else
                    {   
                        idx--;
                    }
                }
            }
            for(size_t i = 0; i< strs.size(); i++)
            {
                result += strs[i];
            }

            return result;
        }
};

 

[LeetCode]ZigZag Conversion

标签:

原文地址:http://www.cnblogs.com/diegodu/p/4250016.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!