题目

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

Write the code that will take a string and make this conversion given a number of rows:

string convert(string text, int nRows);

原题地址

解题思路

这个是个纯粹找规律的题，其他没啥特殊的。下面的例子nRows=4；

找规律按照数组小标开始，寻找下标出现的规律，

1. 第一行和最后一行相邻元素下标之差为 2*nRows-2;

2. 除过第一行和最后一行，其余行要多一个元素，该元素出现的下标和行号有关，比如5 = 1 + 6 - 2，可以总结出规律为 j + 2*nRows-2 - 2*i;

关于 i 和 j 看以看下面的代码。

代码实现

class Solution {
public:
    string convert(string s, int nRows) {
        if(nRows <= 1) return s;
        int base = 2*nRows-2;
        string ret;
        int n = s.size();
        for(int i=0; i<nRows; ++i){
            for(int j=i; ; j += base){
                if(j>=n) break;
                ret.append(1,s[j]);
                if(i==0 || i==nRows-1) continue;
                int temp = j + base - 2*i;
                if(temp <s.size()){
                    ret.append(1,s[temp]);
                }
            }
        }
        return ret;
    }
};