力扣 6. Z 字形变换

题目来源:https://leetcode-cn.com/problems/zigzag-conversion/

大致题意:
给一个字符串和行数 r,把字符串按照从上到下、从左到右进行 Z 字形排列,然后对排列后的字符串按照从左至右、从上到下的顺序取出、

思路

直接模拟变换过程

按照题目要求,在 Z 变换过程中,会先向下填 r 个字符,再向右上填 r - 2 个字符,这个过程可以视为一个周期 t = r * 2 - 2,那么一共有 n / t (向上取整,最后一个周期可能不完整,但是不影响)个周期

求出周期后,就可以按照周期变化规则进行模拟,为了节省内存,可以直接使用 StringBuffer 表示每行对应的字符串

模拟

  1. 求出周期 t
  2. 遍历整个字符串,索引为 i。初始时设行标记 x = 0。那么当 i mod t < r - 1 时,需要继续向下填字符,行数 x + 1;否则要向右上填字符,行数 x - 1
  3. 每次将索引 i 处的字符放入行标记 x 对应的 StringBuffer 对象末尾,然后根据当前索引 i 判断下一次应该放的行数

代码:

public String convert(String s, int numRows) {
        int n = s.length();
        int r = numRows;
        // 若行数为 1 或者大于字符串长度,直接返回原字符串即可
        if (r == 1 || r >= n) {
            return s;
        }
        // 周期
        int t = r * 2 - 2;
        // 行对应的字符串对象
        StringBuffer[] sb = new StringBuffer[r];
        for (int i = 0; i < r; i++) {
            sb[i] = new StringBuffer();
        }
        // 遍历
        for (int i = 0, x = 0; i < n; i++) {
            // 将当前字符放入对应行
            sb[x].append(s.charAt(i));
            // 判断下次放字符的行数
            if (i % t < r - 1) {
                x++;
            } else {
                x--;
            }
        }
        StringBuffer ans = new StringBuffer();
        for (int i = 0; i < r; i++) {
            ans.append(sb[i]);
        }
        return ans.toString();
    }
上一篇:[LeetCode] 1706. Where Will the Ball Fall


下一篇:排序算法