-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpartial_window.py
88 lines (68 loc) · 3.07 KB
/
partial_window.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
# There is a bookstore owner that has a store open for n minutes. Every minute, some number of customers enter the store. You are given an integer array customers of length n where customers[i] is the number of the customer that enters the store at the start of the ith minute and all those customers leave after the end of that minute.
# On some minutes, the bookstore owner is grumpy. You are given a binary array grumpy where grumpy[i] is 1 if the bookstore owner is grumpy during the ith minute, and is 0 otherwise.
# 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 minutes consecutive minutes, 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], minutes = 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.
# Example 2:
# Input: customers = [1], grumpy = [0], minutes = 1
# Output: 1
# here is brute-force approach for above solution
class Solution(object):
def maxSatisfied(self, customers, grumpy, minutes):
"""
:type customers: List[int]
:type grumpy: List[int]
:type minutes: int
:rtype: int
"""
max_profit=0
start=0
last=start+minutes
change_first=start
change_last=last
n=len(customers)
result=0
while last<=n:
total_result=0
for i in range(n):
if(grumpy[i]==0):
total_result+=customers[i]
else:
if(i>=start and i<last):
total_result+=customers[i]
if(total_result>max_profit):
max_profit=total_result
start+=1
last+=1
return max_profit
#here is best approach for this one
class Solution:
def maxSatisfied(self, customers, grumpy, minutes):
total = 0
ans = 0
# Calculate total satisfied customers when not grumpy
for i in range(len(customers)):
total += (1 - grumpy[i]) * customers[i]
window_all = 0
partial_window = 0
for i in range(len(customers)):
window_all += customers[i]
partial_window += (1 - grumpy[i]) * customers[i]
if i + 1 >= minutes:
ans = max(ans, total - partial_window + window_all)
left = i - minutes + 1
window_all -= customers[left]
partial_window -= (1 - grumpy[left]) * customers[left]
return ans
# Test case
sol = Solution()
customers = [1, 0, 1, 2, 1, 1, 7, 5]
grumpy = [0, 1, 0, 1, 0, 1, 0, 1]
minutes = 3
result = sol.maxSatisfied(customers, grumpy, minutes)
print("Result:", result) # Expected output: 16