leetcode题解-6. ZigZag Conversion

最新推荐文章于 2022-04-17 19:49:20 发布

原创最新推荐文章于 2022-04-17 19:49:20 发布 · 662 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode题解专栏收录该内容

63 篇文章

订阅专栏

博客介绍了将字符串按锯齿形排列，再按行顺序输出组成新字符串的问题。给出具体例子，如输入特定字符串和行数，得到相应输出。还分析了解题思路，即先加往下走列的字符，再加往上走的字符，时间和空间复杂度均为O(n)。

题意：将字符串按照锯齿形排列，然后把从上到下把所有行顺序输出，组成新的字符串。

例子：
Input: s = “PAYPALISHIRING”, numRows = 3
Output: “PAHNAPLSIIGYIR”
这里写图片描述

Example 2:

Input: s = “PAYPALISHIRING”, numRows = 4
Output: “PINALSIGYAHRPI”
Explanation:
这里写图片描述

分析：对于每一行先把往下走的那一列的字符加进去，然后有往上走的字符再加进去即可。时间复杂度是O(n),空间复杂度是O(n),代码如下：

class Solution {
    public String convert(String s, int numRows) {
        if(numRows == 1) return s;
        ArrayList<StringBuilder> sbArr = new ArrayList<>();
        for(int i = 0; i < Math.min(numRows, s.length()); i++){
            StringBuilder sb = new StringBuilder();
            sbArr.add(sb);
        }
        // 用来统计当前是第几行
        int rowCount = 0;
        // 用来统计是向下查还是向上查
        int downCount = 1;

        for(int i = 0; i < s.length(); i++){
            if(downCount == 1){
                sbArr.get(rowCount).append(s.charAt(i));
                if(rowCount < numRows - 1){
                    rowCount++;
                }else{
                    downCount = 1 - downCount;
                    rowCount--;
                }
            }else{
                sbArr.get(rowCount).append(s.charAt(i));
                if(rowCount > 0){
                    rowCount--;
                }else{
                    downCount = 1 - downCount;
                    rowCount++;
                }
            }
        }
        StringBuilder result = new StringBuilder();
        for(StringBuilder cStr : sbArr){
            result.append(cStr);
        }
        return result.toString();

    }
}