题目描述

将字符串 “PAYPALISHIRING” 以Z字形排列成给定的行数：（下面这样的形状）

P   A   H   N
A P L S I I G
Y   I   R

之后按逐行顺序依次排列："PAHNAPLSIIGYIR"

实现一个将字符串进行指定行数的转换的函数:

string convert(string text, int nRows);

convert("PAYPALISHIRING", 3) 应当返回 "PAHNAPLSIIGYIR" 。

思路

以3行为例numRows = 3

0   4   8     12
1 3 5 7 9  11 13
2   6   10

4个数为一个循环，即circle = numRows * 2 - 2;
遍历3行for (int j = 0; j < numRows; j++)
第一行余4为0
第二行余4为1或3
第三行余4为2

代码

public String convert(String s, int numRows) {
    if (numRows == 1) return s;
    int circle = numRows * 2 -2;
    String res = "";
    for (int j = 0; j < numRows; j++) {
        for (int i = 0; i < s.length(); i++) {
        if (i % circle == j || i % circle == circle - j) res += s.charAt(i);
        }
    }
    return res;
}