【leetcode】ZigZag Conversion

ZigZag Conversion

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

 

ZigZag的意思：

 

一： 2排的时候，1到n的排序

1 3 5 7 9  。。。

2 4 6 8 10 。。。

 

 

二： 3排的时候，1到n的排序

1   5   9  。。。

2 4 6 8 10 。。。

3   7   11 。。。

 

三： 4排的时候，1到n的排序

1    7       13 。。。 

2  6 8    12 14 。。。

3 5  9  11   15 。。。

4    10      16 。。。

 

对于第一行（0）和最后一行（nRows-1），水平方向两个元素的间隔为：2（nRows-1）

对于第i行来说，水平方向的两个元素间隔为2（nRows-1-i），2（i）

 

 

class Solution {
public:
    string convert(string s, int nRows) {       
        if(nRows==1)
        {
            return s;
        }
        string result;
        int n=s.length();
        int step;
        bool flag;
        for(int i=0;i<nRows;i++)
        {
            int j=i;
            flag=false;
            while(j<n)
            {
                result.push_back(s[j]);
                if(i==0||i==nRows-1)
                {
                    step=2*(nRows-1);
                }
                else
                {
                    if(flag==false)
                    {
                        step=2*(nRows-1-i);
                        flag=true;
                    }
                    else
                    {
                        step=2*i;
                        flag=false;
                    }
                }
                j+=step;
            }
        }       
        return result;
    }
};