leetcode6.Z 字形转换

个人题解与官方题解二思路相同，但官方题解的代码表达更简洁。

题目

链接：https://leetcode-cn.com/problems/zigzag-conversion

官方题解一:按行排序

思路

通过从左向右迭代字符串，我们可以轻松地确定字符位于 Z 字形图案中的哪一行。

算法

我们可以使用 min(numRows,len(s))个列表来表示 Z 字形图案中的非空行。

从左到右迭代 s，将每个字符添加到合适的行。可以使用当前行和当前方向这两个变量对合适的行进行跟踪。

只有当我们向上移动到最上面的行或向下移动到最下面的行时，当前方向才会发生改变。

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

        if (numRows == 1) return s;

        List<StringBuilder> rows = new ArrayList<>();
        for (int i = 0; i < Math.min(numRows, s.length()); i++)
            rows.add(new StringBuilder());

        int curRow = 0;
        boolean goingDown = false;

        for (char c : s.toCharArray()) {
            rows.get(curRow).append(c);
            if (curRow == 0 || curRow == numRows - 1) goingDown = !goingDown;
            curRow += goingDown ? 1 : -1;
        }

        StringBuilder ret = new StringBuilder();
        for (StringBuilder row : rows) ret.append(row);
        return ret.toString();
    }
}

复杂度分析

时间复杂度：O(n)，其中 n==len(s)
空间复杂度：O(n)

作者：LeetCode
链接：https://leetcode-cn.com/problems/zigzag-conversion/solution/z-zi-xing-bian-huan-by-leetcode/
来源：力扣（LeetCode）

代码（带注释）

import java.util.ArrayList;
import java.util.List;

/**
 * leetcode-6-middle
 */
public class zigzag_conversion {
    static class Solution {
        /**
         * 个人题解
         * 思路：1、同一行的竖列上的字符之间间隔和斜列上的字符之间的间隔均为gap，依次拼接成结果字符串；
         * 		 2、第一行和最后一行只有竖列没有斜列
         */
        String convert(String s, int numRows) {
            if(numRows >= s.length() || numRows == 1) return s;
            int gap = 2 * numRows - 2;                 //每列两个对应元素的间隔
            StringBuilder ans = new StringBuilder();   //存储结果值的StringBuilder对象
            for(int i = 1; i <= numRows; i++){         //对每行进行遍历
                int m = i;                             //java中每次循环不会重置i的值，所以内部对i的值进行修改的话会影响循环控制
                if( m == 1 || m == numRows){           //第一行和最后一行只需根据第一个元素的位置依次累加gap即可
                    while (m <= s.length()) {
                        ans.append(s.charAt(--m));
                        m += gap + 1;//因为检索时减去了1所以需要加回来
                    }
                    continue;
                }

                int j = 2 * numRows - m;                // j用来表示处于两竖列之间的元素的位置
                while(m <= s.length() || j <= s.length()){
                    if(m <= s.length()) {
                        ans.append(s.charAt(--m));
                        m += gap + 1;
                    }
                    if(j <= s.length()) {
                        ans.append(s.charAt(--j));
                        j += gap + 1;
                    }
                }
            }

            return ans.toString();
        }

        /**
         * 官方题解二，思路与上同，但表达更简洁
         * @param s：需要转换的字符串
         * @param numRows ： 指定Z字形的最大行数
         * @return 返回Z字形转换后按行重组的字符串
         */
        String convertBetter(String s, int numRows) {
            
            if (numRows == 1) return s;

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

            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));     //根据竖列上的字符与同行相邻的斜列上的字符的数量关系，得出j的位置
                }
            }
            return ret.toString();
        }

        /**
         * 官方题解一
         * 思路： 沿着Z字形依次遍历，将每个字符放入其行对应的StringBuilder对象中
         */
        String convertA(String s, int numRows) {
            if (numRows == 1 || numRows >= s.length()) return s;

            List<StringBuilder> rows = new ArrayList<>();    //存储用以顺序每行字符的StringBuilder对象
            for (int i = 0; i < Math.min(numRows, s.length()); i++)    //初始化
                rows.add(new StringBuilder());

            int curRow = 0;                //从第一行开始
            boolean goingDown = false;     //用以表示当前方向是向上遍历还是向下遍历，由于每次首行和尾行需要置否所以初值为

            for (char c : s.toCharArray()) {
                rows.get(curRow).append(c);
                if (curRow == 0 || curRow == numRows - 1) goingDown = !goingDown;   //第一行和最后一行需要改变方向
                curRow += goingDown ? 1 : -1;   //正在向下走则行数递增，向上走行数递减
            }

            StringBuilder ret = new StringBuilder();
            for (StringBuilder row : rows) ret.append(row);
            return ret.toString();
        }
    }
    
}