mirror of
https://github.com/halfrost/LeetCode-Go.git
synced 2025-07-18 11:24:52 +08:00
添加 problem 1052
This commit is contained in:
@ -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 的节点。这两个节点之间都是算宽度的。最终输出最大的宽度即可。此题的关键在于如何有效的找到每一层的左右边界。
|
- 这一题可以用 BFS 也可以用 DFS,但是用 BFS 比较方便。按照层序遍历,依次算出每层最左边不为 `null` 的节点和最右边不为 `null` 的节点。这两个节点之间都是算宽度的。最终输出最大的宽度即可。此题的关键在于如何有效的找到每一层的左右边界。
|
||||||
- 这一题可能有人会想着先补全满二叉树,然后每层分别找左右边界。这种方法提交以后会卡在 104 / 108 这组测试用例上,这组测试用例会使得最后某几层填充出现的满二叉树节点特别多,最终导致 Memory Limit Exceeded 了。
|
- 这一题可能有人会想着先补全满二叉树,然后每层分别找左右边界。这种方法提交以后会卡在 `104 / 108` 这组测试用例上,这组测试用例会使得最后某几层填充出现的满二叉树节点特别多,最终导致 `Memory Limit Exceeded` 了。
|
||||||
- 由于此题要找每层的左右边界,实际上每个节点的 Val 值是我们不关心的,那么可以把这个值用来标号,标记成该节点在每层中的序号。父亲节点在上一层中的序号是 x,那么它的左孩子在下一层满二叉树中的序号是 2\*x,它的右孩子在下一层满二叉树中的序号是 2\*x + 1。将所有节点都标上号,用 BFS 层序遍历每一层,每一层都找到左右边界,相减拿到宽度,动态维护最大宽度,就是本题的最终答案。
|
- 由于此题要找每层的左右边界,实际上每个节点的 `Val` 值是我们不关心的,那么可以把这个值用来标号,标记成该节点在每层中的序号。父亲节点在上一层中的序号是 x,那么它的左孩子在下一层满二叉树中的序号是 `2*x`,它的右孩子在下一层满二叉树中的序号是 `2*x + 1`。将所有节点都标上号,用 BFS 层序遍历每一层,每一层都找到左右边界,相减拿到宽度,动态维护最大宽度,就是本题的最终答案。
|
@ -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
|
||||||
|
}
|
@ -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")
|
||||||
|
}
|
45
Algorithms/1052. Grumpy Bookstore Owner/README.md
Executable file
45
Algorithms/1052. Grumpy Bookstore Owner/README.md
Executable file
@ -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`。
|
Reference in New Issue
Block a user