LeetCode破解之蜡烛盘子

99 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第23天,点击查看活动详情

题目描述

给你一个长桌子,桌子上盘子和蜡烛排成一列。给你一个下标从 0 开始的字符串 s ,它只包含字符 '' 和 '|' ,其中 '' 表示一个 盘子 ,'|' 表示一支 蜡烛 。

同时给你一个下标从 0 开始的二维整数数组 queries ,其中 queries[i] = [lefti, righti] 表示 子字符串 s[lefti...righti] (包含左右端点的字符)。对于每个查询,你需要找到 子字符串中 在 两支蜡烛之间 的盘子的 数目 。如果一个盘子在 子字符串中 左边和右边 都 至少有一支蜡烛,那么这个盘子满足在 两支蜡烛之间 。

  • 比方说,s = "|||||" ,查询 [3, 8] ,表示的是子字符串 "||**|" 。子字符串中在两支蜡烛之间的盘子数目为 2 ,子字符串中右边两个盘子在它们左边和右边 都 至少有一支蜡烛。

请你返回一个整数数组 answer ,其中 answer[i] 是第 i 个查询的答案。

示例 1:

输入:s = "||***|", queries = [[2,5],[5,9]] 输出:[2,3] 解释:

  • queries[0] 有两个盘子在蜡烛之间。
  • queries[1] 有三个盘子在蜡烛之间。

前缀和

思路分析:首先我们先预算出每个位置的星号数量,做好数据缓存,再来计算结束位置,通过计算的每个位置,对应的应该是最靠近的左边的柱子编号。最后;唉计算起始位置,计算每个位置,对应的应该是最靠近的右边的柱子编号,那么把两个位置,对应星号数量相减,就可以得出我们想要的结果。解析点:

  • 记录每个位置在它左/右离它最近的盘子下标=> l[i], r[i]统计每个位置及之前有多少盘子q。
  • 对于给定区间[a, b] 盘子之间的蜡烛为 q[l[i]] - q[r[i]];

实现步骤:

  • 为每一个盘子定义前缀数组,和后缀数组。如果盘子的前缀为true && 盘子的后缀数组也为true 说明该盘子符合count++;
  • 前缀数组:left[i]:如果是盘子记录它前面有蜡烛的位置。有left[i]=i;
  • 后缀数组:right[i]:如果是盘子记录它后面有蜡烛的位置。有right[i]=i
class Solution {
    public int[] platesBetweenCandles(String s, int[][] q) {
        int n = s.length();
        int m = q.length;
        int[] left = new int[n];
        int[] right = new int[n];
        int[] sum = new int[n];
        int[] ans = new int[m];
        int index = -1;
        int cnt = 0;
        for (int i = 0; i < n; i++) {
            if (s.charAt(i) == '|') {
                index = i;
                cnt++;
            }
            left[i] = index;
            sum[i] = cnt;
        }

        index = -1;
        for (int i = n - 1; i >= 0; i--) {
            if (s.charAt(i) == '|') {
                index = i;
            }
            right[i] = index;
        }

        for (int i = 0; i < m; i++) {
            int x = q[i][0];
            int y = q[i][1];
            int l = right[x];
            int r = left[y];
            if (l >= 0 && r >= 0 && l < r) {
                ans[i] = r - l - (sum[r] - sum[l]);
            }
        }
        return ans;
    }
}

最后

这道题目常规的方式就是通过前缀和来做,不然容易超时,也可以再搭配上预处理,HashMap,二分等方法来优化,动态规划也可以解题。