-
Notifications
You must be signed in to change notification settings - Fork 0
/
number-of-students-unable-to-eat-lunch_1700.py
69 lines (53 loc) · 2.91 KB
/
number-of-students-unable-to-eat-lunch_1700.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
# The school cafeteria offers circular and square sandwiches at lunch break, referred to by numbers 0 and 1 respectively.
# All students stand in a queue. Each student either prefers square or circular sandwiches.
# The number of sandwiches in the cafeteria is equal to the number of students. The sandwiches are placed in a stack. At each step:
# If the student at the front of the queue prefers the sandwich on the top of the stack, they will take it and leave the queue.
# Otherwise, they will leave it and go to the queue's end.
# This continues until none of the queue students want to take the top sandwich and are thus unable to eat.
# You are given two integer arrays students and sandwiches where sandwiches[i] is the type of the ith sandwich in the stack
# (i = 0 is the top of the stack) and students[j] is the preference of the jth student in the initial queue
# (j = 0 is the front of the queue). Return the number of students that are unable to eat.
# Example 1:
# Input: students = [1,1,0,0], sandwiches = [0,1,0,1]
# Output: 0
# Explanation:
# - Front student leaves the top sandwich and returns to the end of the line making students = [1,0,0,1].
# - Front student leaves the top sandwich and returns to the end of the line making students = [0,0,1,1].
# - Front student takes the top sandwich and leaves the line making students = [0,1,1] and sandwiches = [1,0,1].
# - Front student leaves the top sandwich and returns to the end of the line making students = [1,1,0].
# - Front student takes the top sandwich and leaves the line making students = [1,0] and sandwiches = [0,1].
# - Front student leaves the top sandwich and returns to the end of the line making students = [0,1].
# - Front student takes the top sandwich and leaves the line making students = [1] and sandwiches = [1].
# - Front student takes the top sandwich and leaves the line making students = [] and sandwiches = [].
# Hence all students are able to eat.
# Example 2:
# Input: students = [1,1,1,0,0,1], sandwiches = [1,0,0,0,1,1]
# Output: 3
# ---------------------------------------Runtime 52 ms Beats 26.56% Memory 16.3 MB Beats 27.51%---------------------------------------
from typing import List
# My solution
class Solution:
def countStudents(self, students: List[int], sandwiches: List[int]) -> int:
i = j = 0
while students:
if students[i] == sandwiches[j]:
students.pop(i)
sandwiches.pop(j)
elif sandwiches[j] not in students:
return len(students)
else:
students.append(students.pop(i))
return len(students)
# Solution @lee215
import collections
# Complexity
# Time O(n)
# Space O(2)
class Solution:
def countStudents(self, A, B):
count = collections.Counter(A)
n, k = len(A), 0
while k < n and count[B[k]]:
count[B[k]] -= 1
k += 1
return n - k