From 75df795ecbc8c51dd1d2263b60b7b8d30916b289 Mon Sep 17 00:00:00 2001 From: YDZ Date: Sat, 24 Aug 2019 13:19:44 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=20problem=201052?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../README.md | 6 +-- .../1052. Grumpy Bookstore Owner.go | 53 ++++++++++++++++++ .../1052. Grumpy Bookstore Owner_test.go | 54 +++++++++++++++++++ .../1052. Grumpy Bookstore Owner/README.md | 45 ++++++++++++++++ 4 files changed, 155 insertions(+), 3 deletions(-) create mode 100644 Algorithms/1052. Grumpy Bookstore Owner/1052. Grumpy Bookstore Owner.go create mode 100644 Algorithms/1052. Grumpy Bookstore Owner/1052. Grumpy Bookstore Owner_test.go create mode 100755 Algorithms/1052. Grumpy Bookstore Owner/README.md diff --git a/Algorithms/0662. Maximum Width of Binary Tree/README.md b/Algorithms/0662. Maximum Width of Binary Tree/README.md index d3c35c1e..a158b5ac 100755 --- a/Algorithms/0662. Maximum Width of Binary Tree/README.md +++ b/Algorithms/0662. Maximum Width of Binary Tree/README.md @@ -77,6 +77,6 @@ The width of one level is defined as the length between the end-nodes (the leftm - 给出一个二叉树，求这棵树最宽的部分。 -- 这一题可以用 BFS 也可以用 DFS，但是用 BFS 比较方便。按照层序遍历，依次算出每层最左边不为 null 的节点和最右边不为 null 的节点。这两个节点之间都是算宽度的。最终输出最大的宽度即可。此题的关键在于如何有效的找到每一层的左右边界。 -- 这一题可能有人会想着先补全满二叉树，然后每层分别找左右边界。这种方法提交以后会卡在 104 / 108 这组测试用例上，这组测试用例会使得最后某几层填充出现的满二叉树节点特别多，最终导致 Memory Limit Exceeded 了。 -- 由于此题要找每层的左右边界，实际上每个节点的 Val 值是我们不关心的，那么可以把这个值用来标号，标记成该节点在每层中的序号。父亲节点在上一层中的序号是 x，那么它的左孩子在下一层满二叉树中的序号是 2\*x，它的右孩子在下一层满二叉树中的序号是 2\*x + 1。将所有节点都标上号，用 BFS 层序遍历每一层，每一层都找到左右边界，相减拿到宽度，动态维护最大宽度，就是本题的最终答案。 +- 这一题可以用 BFS 也可以用 DFS，但是用 BFS 比较方便。按照层序遍历，依次算出每层最左边不为 `null` 的节点和最右边不为 `null` 的节点。这两个节点之间都是算宽度的。最终输出最大的宽度即可。此题的关键在于如何有效的找到每一层的左右边界。 +- 这一题可能有人会想着先补全满二叉树，然后每层分别找左右边界。这种方法提交以后会卡在 `104 / 108` 这组测试用例上，这组测试用例会使得最后某几层填充出现的满二叉树节点特别多，最终导致 `Memory Limit Exceeded` 了。 +- 由于此题要找每层的左右边界，实际上每个节点的 `Val` 值是我们不关心的，那么可以把这个值用来标号，标记成该节点在每层中的序号。父亲节点在上一层中的序号是 x，那么它的左孩子在下一层满二叉树中的序号是 `2*x`，它的右孩子在下一层满二叉树中的序号是 `2*x + 1`。将所有节点都标上号，用 BFS 层序遍历每一层，每一层都找到左右边界，相减拿到宽度，动态维护最大宽度，就是本题的最终答案。 \ No newline at end of file diff --git a/Algorithms/1052. Grumpy Bookstore Owner/1052. Grumpy Bookstore Owner.go b/Algorithms/1052. Grumpy Bookstore Owner/1052. Grumpy Bookstore Owner.go new file mode 100644 index 00000000..a0079f33 --- /dev/null +++ b/Algorithms/1052. Grumpy Bookstore Owner/1052. Grumpy Bookstore Owner.go @@ -0,0 +1,53 @@ +package leetcode + +// 解法一 滑动窗口优化版 +func maxSatisfied(customers []int, grumpy []int, X int) int { + customer0, customer1, maxCustomer1, left, right := 0, 0, 0, 0, 0 + for ; right < len(customers); right++ { + if grumpy[right] == 0 { + customer0 += customers[right] + } else { + customer1 += customers[right] + for right-left+1 > X { + if grumpy[left] == 1 { + customer1 -= customers[left] + } + left++ + } + if customer1 > maxCustomer1 { + maxCustomer1 = customer1 + } + } + } + return maxCustomer1 + customer0 +} + +// 解法二 滑动窗口暴力版 +func maxSatisfied1(customers []int, grumpy []int, X int) int { + left, right, res := 0, -1, 0 + for left < len(customers) { + if right+1 < len(customers) && right-left < X-1 { + right++ + } else { + if right-left+1 == X { + res = max(res, sumSatisfied(customers, grumpy, left, right)) + } + left++ + } + } + return res +} + +func sumSatisfied(customers []int, grumpy []int, start, end int) int { + sum := 0 + for i := 0; i < len(customers); i++ { + if i < start || i > end { + if grumpy[i] == 0 { + sum += customers[i] + } + } else { + sum += customers[i] + } + } + return sum +} diff --git a/Algorithms/1052. Grumpy Bookstore Owner/1052. Grumpy Bookstore Owner_test.go b/Algorithms/1052. Grumpy Bookstore Owner/1052. Grumpy Bookstore Owner_test.go new file mode 100644 index 00000000..327c5875 --- /dev/null +++ b/Algorithms/1052. Grumpy Bookstore Owner/1052. Grumpy Bookstore Owner_test.go @@ -0,0 +1,54 @@ +package leetcode + +import ( + "fmt" + "testing" +) + +type question1052 struct { + para1052 + ans1052 +} + +// para 是参数 +// one 代表第一个参数 +type para1052 struct { + customers []int + grumpy []int + x int +} + +// ans 是答案 +// one 代表第一个答案 +type ans1052 struct { + one int +} + +func Test_Problem1052(t *testing.T) { + + qs := []question1052{ + + question1052{ + para1052{[]int{4, 10, 10}, []int{1, 1, 0}, 2}, + ans1052{24}, + }, + + question1052{ + para1052{[]int{1}, []int{0}, 1}, + ans1052{1}, + }, + + question1052{ + para1052{[]int{1, 0, 1, 2, 1, 1, 7, 5}, []int{0, 1, 0, 1, 0, 1, 0, 1}, 3}, + ans1052{16}, + }, + } + + fmt.Printf("------------------------Leetcode Problem 1052------------------------\n") + + for _, q := range qs { + _, p := q.ans1052, q.para1052 + fmt.Printf("【input】:%v 【output】:%v\n", p, maxSatisfied(p.customers, p.grumpy, p.x)) + } + fmt.Printf("\n\n\n") +} diff --git a/Algorithms/1052. Grumpy Bookstore Owner/README.md b/Algorithms/1052. Grumpy Bookstore Owner/README.md new file mode 100755 index 00000000..01d3f866 --- /dev/null +++ b/Algorithms/1052. Grumpy Bookstore Owner/README.md @@ -0,0 +1,45 @@ +# [1052. Grumpy Bookstore Owner](https://leetcode.com/problems/grumpy-bookstore-owner/) + + +## 题目: + +Today, the bookstore owner has a store open for `customers.length`minutes. Every minute, some number of customers (`customers[i]`) enter the store, and all those customers leave after the end of that minute. + +On some minutes, the bookstore owner is grumpy. If the bookstore owner is grumpy on the i-th minute, `grumpy[i] = 1`, otherwise `grumpy[i] = 0`. When the bookstore owner is grumpy, the customers of that minute are not satisfied, otherwise they are satisfied. + +The bookstore owner knows a secret technique to keep themselves not grumpy for `X` minutes straight, but can only use it once. + +Return the maximum number of customers that can be satisfied throughout the day. + +**Example 1:** + + Input: customers = [1,0,1,2,1,1,7,5], grumpy = [0,1,0,1,0,1,0,1], X = 3 + Output: 16 + Explanation: The bookstore owner keeps themselves not grumpy for the last 3 minutes. + The maximum number of customers that can be satisfied = 1 + 1 + 1 + 1 + 7 + 5 = 16. + +**Note:** + +- `1 <= X <= customers.length == grumpy.length <= 20000` +- `0 <= customers[i] <= 1000` +- `0 <= grumpy[i] <= 1` + + +## 题目大意 + +今天，书店老板有一家店打算试营业 customers.length 分钟。每分钟都有一些顾客（customers[i]）会进入书店，所有这些顾客都会在那一分钟结束后离开。在某些时候，书店老板会生气。 如果书店老板在第 i 分钟生气，那么 grumpy[i] = 1，否则 grumpy[i] = 0。 当书店老板生气时，那一分钟的顾客就会不满意，不生气则他们是满意的。书店老板知道一个秘密技巧，能抑制自己的情绪，可以让自己连续 X 分钟不生气，但却只能使用一次。请你返回这一天营业下来，最多有多少客户能够感到满意的数量。 + +提示： + +1. 1 <= X <= customers.length == grumpy.length <= 20000 +2. 0 <= customers[i] <= 1000 +3. 0 <= grumpy[i] <= 1 + + + +## 解题思路 + + +- 给出一个顾客入店时间表和书店老板发脾气的时间表。两个数组的时间是一一对应的，即相同下标对应的相同的时间。书店老板可以控制自己在 X 分钟内不发火，但是只能控制一次。问有多少顾客能在书店老板不发火的时候在书店里看书。抽象一下，给出一个价值数组和一个装着 0 和 1 的数组，当价值数组的下标对应另外一个数组相同下标的值是 0 的时候，那么这个价值可以累加，当对应是 1 的时候，就不能加上这个价值。现在可以让装着 0 和 1 的数组中连续 X 个数都变成 0，问最终价值最大是多少？ +- 这道题是典型的滑动窗口的题目。最暴力的解法是滑动窗口右边界，当与左边界的距离等于 X 的时候，计算此刻对应的数组的总价值。当整个宽度为 X 的窗口滑过整个数组以后，输出维护的最大值即可。这个方法耗时比较长。因为每次计算数组总价值的时候都要遍历整个数组。这里是可以优化的地方。 +- 每次计算数组总价值的时候，其实目的是为了找到宽度为 X 的窗口对应里面为 1 的数累加和最大，因为如果把这个窗口里面的 1 都变成 0 以后，那么对最终价值的影响也最大。所以用一个变量 `customer0` 专门记录脾气数组中为 0 的对应的价值，累加起来。因为不管怎么改变，为 0 的永远为 0，唯一变化的是 1 变成 0 。用 `customer1` 专门记录脾气数组中为 1 的对应的价值。在窗口滑动过程中找到 `customer1` 的最大值。最终要求的最大值就是 `customer0 + maxCustomer1`。