ZigZag Conversion

来源:互联网 发布:csolfps优化 编辑:程序博客网 时间:2024/06/05 19:02

The string "PAYPALISHIRING" iswritten 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 n Rows);

convert("PAYPALISHIRING", 3) should return "PAHNAPLSIIGYIR".

思路:主要需要理清楚zigzag的形成,可以多举几个列子,n=1n=2n=3n=4的时候。

例如n=4的时候,如上图所示,紫色框中是一个zigzagzigSize = 2 * n -2

红色部分的数字是可以有规律的计算出来的,每一行的差为zigSize大小,主要是黑色部分的计算,int middleIdx = base + zigSize - 2 * i;注意首尾行与其他行不同。


class Solution {public:    string convert(string s, int nRows) {        if (nRows <= 1 || s.size() <= nRows) {            return s;        }                string result;        int zigSize = 2 * nRows - 2;                for (int i = 0; i < nRows; i++) {            for (int base = i;; base = base + zigSize) {                if(base >= s.size()) {                    break;                }                char tmp = s[base];                result.push_back(tmp);                if (i > 0 && i < nRows - 1) {                    int middleIdx = base + zigSize - 2 * i;                    if (middleIdx < s.size()) {                        char tmp = s[middleIdx];                        result.push_back(tmp);                    }                }            }        }        return result;    }};


0 0
原创粉丝点击