Leetcode 06：Z字形变换

题目描述：

将一个给定字符串 s 根据给定的行数 numRows ，以从上往下、从左到右进行 Z 字形排列。

比如输入字符串为 "PAYPALISHIRING" 行数为 3 时，排列如下：

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

之后，你的输出需要从左往右逐行读取，产生出一个新的字符串，比如："PAHNAPLSIIGYIR"。

请你实现这个将字符串进行指定行数变换的函数：

string convert(string s, int numRows);

示例 1：

输入：s = "PAYPALISHIRING", numRows = 3
输出："PAHNAPLSIIGYIR"

示例 2：

输入：s = "PAYPALISHIRING", numRows = 4
输出："PINALSIGYAHRPI"

解释：
P     I    N
A   L S  I G
Y A   H R
P     I

示例 3：

输入：s = "A", numRows = 1
输出："A"

说明：所谓的Z字形转换，更多的人认为应该是N字形转换，如下：

 

解法一：按行排序

思路：从左到右遍历字符串，之后确定当前字符位于Z字形图案的哪一行。假设此时的numRows=3，在移动过程中，需要两个变量：当前行和当前方向，且只有移动到第一行和最后一行的时候，才需要转变方向。最后再将每一行的拼接到一起即可。如下图：

 代码：

class Solution {
    public String convert(String s, int numRows) {
        //如果小于等于1，直接返回s
        if(numRows <= 1){
            return s;
        }
        List<StringBuffer> list = new ArrayList<>();
        //当前所处的行
        int curRow = 0;
        //判断是否到达第一行或者最后一行
        boolean flag = false;
        //按照行数利用StringBuffer创建对象
        for (int i = 0; i < Math.min(numRows, s.length()); i++) {
            list.add(new StringBuffer());
        }
        //遍历字符串
        for (char ch : s.toCharArray()) {
            //根据下标获取到当前行的StringBuffer对象
            list.get(curRow).append(ch);
            //如果是第一行或者最后一行，就让标志位取反
            if(curRow == 0 || curRow == numRows-1){
                flag = !flag;
            }
            //如果当前行向下走则行数+1，向上走就-1
            curRow = flag ? curRow + 1 : curRow - 1;
        }
        //拼接字符串
        StringBuffer result= new StringBuffer();
        for (StringBuffer sb : list){
            result.append(sb);
        }
        return result.toString();
    }
}

解法二：按行访问。

思路：这种方式更像是找规律。通过分析每行的字符所处的下标位置找到规律，然后依次访问行0、1、2、... 、numRows-1的字符即可。因此，通过分析，对于k（取值范围从0开始）：

（1）行0中的字符处于索引为  k(2 * numRows - 2)  处。

（2）行numsRows-1中的字符位于索引  k(2 * numrRows - 2) + numRows-1处。

（3）其余行  i  中的字符位于索引  k(2 *  numRows - 2) + i 以及 (k+1)(2 * numRows - 2) - i处。

如下图：

 

代码：

public static 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();
    }