【LeetCode】6.Z字形变换

 

题目描述：

    将字符串 "PAYPALISHIRING" 以Z字形排列成给定的行数：

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

    之后从左往右，逐行读取字符："PAHNAPLSIIGYIR"

    实现一个将字符串进行指定行数变换的函数: string convert(string s, int numRows);

1.按行排序

算法思想：从左向右迭代字符串，确定字符位于Z字形图案中的哪一行哪一列；只有当我们向上移动到最上面的行或向下移动到最下面的行时，当前方向才会发生改变。

class Solution {
    public String convert(String s, int numRows) {

        if (numRows == 1) return s;

        StringBuilder ret = new StringBuilder();
        int n = s.length();
        int cycleLen = 2 * numRows - 2;

        for (int i = 0; i < numRows; i++) {
            for (int j = 0; j + i < n; j += cycleLen) {
                ret.append(s.charAt(j + i));
                if (i != 0 && i != numRows - 1 && j + cycleLen - i < n)
                    ret.append(s.charAt(j + cycleLen - i));
            }
        }
        return ret.toString();
    }
}

2.按行访问

算法思想：按照与逐行读取 Z 字形图案相同的顺序访问字符串，首先访问行0的所有字符，然后行1，行2 ……以此类推；

class Solution {
    public String convert(String s, int numRows) {

        if (numRows == 1) return s;

        StringBuilder ret = new StringBuilder();
        int n = s.length();
        int cycleLen = 2 * numRows - 2;

        for (int i = 0; i < numRows; i++) {
            for (int j = 0; j + i < n; j += cycleLen) {
                ret.append(s.charAt(j + i));
                if (i != 0 && i != numRows - 1 && j + cycleLen - i < n)
                    ret.append(s.charAt(j + cycleLen - i));
            }
        }
        return ret.toString();
    }
}