849. Maximize Distance to Closest Person
Problem Description
In this problem, we are provided with an array seats
that represents a row of seats. Each seat can be either occupied (denoted by 1
) or empty (denoted by 0
). We know that there is at least one empty seat and at least one occupied seat in the array. The objective is to find a seat for Alex where, if he sits down, the distance to the nearest person already seated is as large as possible. The distance is defined as the number of seats between Alex's seat and the closest seated person, and we need to maximize this distance to ensure maximum possible personal space for Alex. Our goal is to calculate and return this maximum distance.
Intuition
The intuition behind the solution is to identify the longest stretch of consecutive empty seats, with consideration of the edge cases where the longest stretch might actually start at the beginning of the row or end at the last seat in the row.
Here are key insights to form a strategy:
-
Identify stretches of empty seats and find the longest stretch. This corresponds to finding the largest gap between two occupied seats.
-
Special consideration for the first and last seats: If the first or last seats are empty, the gap is counted from the edge of the row to the first or last occupied seat respectively. This needs special handling since the "distance" in this case is not halved—it is taken as the entire length of the stretch.
-
Find the longest stretch: To maximize the distance to the nearest person, Alex can sit exactly in the middle of the longest stretch of empty seats. The maximum distance to the closest person would then be half the length of this longest stretch. However, if this stretch is at the beginning or end of the row, he will sit at the first or last available seat, resulting in the distance being the entire length of this stretch.
The approach to solving the problem involves iterating through the seats
array to identify the first and last occupied seats, and in the process, also recording the maximum distance between two consecutive occupied seats. The maximum distance Alex can sit away from the closest person would then be the maximum of the following three values:
- The distance from the start of the row to the first occupied seat.
- The distance from the last occupied seat to the end of the row.
- Half the distance of the longest stretch of empty seats between two occupied seats (since Alex would sit exactly in the middle of this stretch).
This method efficiently calculates the optimal seat for Alex in a single pass through the seats
array, avoiding the need for multiple iterations or complex calculations.
Solution Approach
The solution approach involves a single pass through the row of seats
using a for-loop. During this pass, the code keeps track of the index of the first and last occupied seats, as well as the distance to the last occupied seat seen so far when examining each seat. The following elements contribute to the solution:
Variables Used:
first
: This holds the index of the first occupied seat found in the row. Initially, it is set toNone
.last
: This stores the index of the most recently found occupied seat as we iterate through theseats
. It is also initially set toNone
.d
: This variable maintains the maximum distance between two occupied seats as the iteration progresses. We initialize it to0
.
The Algorithm:
-
Iterate through the
seats
array with the indexi
and the valuec
using theenumerate
function, which provides both the index and value during the iteration. -
Check if the current seat
c
is occupied (i.e.,c
is1
):- If an occupied seat is found and it's not the first one (
last
is notNone
), updated
with the maximum value between the currentd
and the difference between the current indexi
and the index of the last occupied seatlast
. - If this is the first occupied seat found (
first
isNone
), store its index infirst
. - Update
last
with the current indexi
because it is the latest occupied seat.
- If an occupied seat is found and it's not the first one (
-
After completing the iteration, calculate the maximum distance Alex can be from the closest person by taking the maximum of three potential values:
- The distance from the start of the row to the first occupied seat (
first
). - The distance from the last occupied seat to the end of the row (
len(seats) - last - 1
). - Half the distance of the longest stretch of empty seats between two occupied seats (
d // 2
).
- The distance from the start of the row to the first occupied seat (
The Code:
class Solution:
def maxDistToClosest(self, seats: List[int]) -> int:
first = last = None
d = 0
for i, c in enumerate(seats):
if c:
if last is not None:
d = max(d, i - last)
if first is None:
first = i
last = i
return max(first, len(seats) - last - 1, d // 2)
This implementation uses a greedy approach to find the maximum distance in an efficient manner. No additional data structure is required aside from a few variables to keep track of indices and the maximum distance found. Because of this, the space complexity is O(1), and since we are iterating through the array only once, the time complexity is O(n), where n
is the number of seats in the row.
Ready to land your dream job?
Unlock your dream job with a 2-minute evaluator for a personalized learning plan!
Start EvaluatorExample Walkthrough
Imagine we have an array seats
that looks like this: [0, 1, 0, 0, 1, 0, 1, 0, 0, 0]
.
To apply our solution approach, let's walk through the algorithm step by step:
- Initialize:
first = last = None
andd = 0
. - Begin iterating through the
seats
array with indices and values. - At index 0, the seat is empty, so no changes are made.
- At index 1, the seat is occupied so:
- Since this is the first occupied seat we've found, set
first
to the current index (first = 1
). - Update
last
to be the current index as well (last = 1
).
- Since this is the first occupied seat we've found, set
- Continue iterating. Seats 2 and 3 are empty, so no changes.
- At index 4, we encounter another occupied seat.
- Since
last
is notNone
, we calculate the distance from the previous occupied seat at index 1 to the current seat at index 4, which yields a distance of3
. Updated
to be the maximum of0
and3
, sod = 3
. - Update
last
to the current index (last = 4
).
- Since
- Move on to seat 5 which is empty, so no changes.
- At index 6, there is another occupied seat.
- Again, calculate the distance (1) and compare it with
d
. Since1
is less than3
, we do not updated
. - Update
last
to6
.
- Again, calculate the distance (1) and compare it with
- Seats 7, 8, and 9 are empty, so no changes occur during these steps.
Once we’ve finished iterating through the array, we consider the maximum distance from the following:
- The distance from the start of the row to the first occupied seat (
first = 1
). - The distance from the last occupied seat to the end of the row (
len(seats) - last - 1
=10 - 6 - 1
=3
). - Half the distance of the longest stretch of empty seats between two occupied seats (
d // 2
=3 // 2
=1
).
The maximum of these values is 3
, which is accounted for by both the distance from the last occupied seat to the end of the row and from the first seat in the row to the first occupied seat. Alex can choose either seat at the start or the end of the row for the maximum distance to the closest person, hence we return 3
as the result.
Here's the walkthrough reflected in the provided code:
class Solution:
def maxDistToClosest(self, seats: List[int]) -> int:
first = last = None
d = 0
for i, c in enumerate(seats):
if c:
if last is not None:
d = max(d, i - last) # At index 4 and index 1, d becomes 3
if first is None:
first = i # At index 1, first becomes 1
last = i # Updates at indices 1, 4, and 6
return max(first, len(seats) - last - 1, d // 2) # Returns max(1, 3, 1) which is 3
By iterating through the array only once and using a simple set of variables to keep track, we find the maximum distance where Alex can sit with a time complexity of O(n) and a space complexity of O(1).
Solution Implementation
1class Solution:
2 def maxDistToClosest(self, seats: List[int]) -> int:
3 # Initialize the first occupied seat index and last occupied seat index to None
4 first_occupied_seat_index = None
5 last_occupied_seat_index = None
6 max_distance = 0 # Initialize the maximum distance to 0
7
8 # Iterate over each seat and its index in the seats list
9 for i, seat in enumerate(seats):
10 # Check if the seat is occupied
11 if seat == 1:
12 # If there was a previously occupied seat, calculate the distance between
13 # the current occupied seat and the last occupied seat and update max_distance
14 if last_occupied_seat_index is not None:
15 distance_between_seats = i - last_occupied_seat_index
16 max_distance = max(max_distance, distance_between_seats)
17
18 # If this is the first occupied seat we've found, store its index
19 if first_occupied_seat_index is None:
20 first_occupied_seat_index = i
21
22 # Update the last occupied seat index to the current index
23 last_occupied_seat_index = i
24
25 # Calculate the maximum distance by considering the first and last seats as well,
26 # if they are unoccupied, and compare it with the maximum distance found between two occupants.
27 # The distance for the first and last seats is handled differently, hence they are compared separately:
28 # max_distance // 2 is used for the middle section since a person can sit in the middle of two occupied seats.
29 return max(
30 first_occupied_seat_index, # Max distance from the start to the first occupied seat
31 len(seats) - last_occupied_seat_index - 1, # Max distance from the last occupied seat to the end
32 max_distance // 2 # Max distance between two occupied seats divided by 2
33 )
34
35# Note: The provided code assumes that 'List' is properly imported from the 'typing' module.
36# If the code is run as-is, make sure to add: from typing import List
37
1class Solution {
2 public int maxDistToClosest(int[] seats) {
3 // Initialize variables for the first occupied seat and last occupied seat.
4 int firstOccupied = -1, lastOccupied = -1;
5
6 // Initialize the maximum distance to the closest person to 0.
7 int maxDistance = 0;
8
9 // Get the number of seats.
10 int seatCount = seats.length;
11
12 // Iterate over each seat.
13 for (int i = 0; i < seatCount; ++i) {
14
15 // Check if the current seat is occupied.
16 if (seats[i] == 1) {
17
18 // Update the maximum distance if there is a last occupied seat to compare with.
19 if (lastOccupied != -1) {
20 maxDistance = Math.max(maxDistance, i - lastOccupied);
21 }
22
23 // If it's the first occupied seat encountered, record its index.
24 if (firstOccupied == -1) {
25 firstOccupied = i;
26 }
27
28 // Update the last occupied seat index.
29 lastOccupied = i;
30 }
31 }
32
33 // Calculate the distance from the first occupied seat and the last seat.
34 int distanceFromStart = firstOccupied;
35 int distanceFromEnd = seatCount - lastOccupied - 1;
36
37 // Get the maximum distance from start, end, and between occupied seats.
38 // The maximum distance in between seats is divided by 2 since it's between two people.
39 return Math.max(maxDistance / 2, Math.max(distanceFromStart, distanceFromEnd));
40 }
41}
42
1#include <vector>
2#include <algorithm> // For using max()
3
4using std::vector;
5using std::max;
6
7class Solution {
8public:
9 // Finds the maximal distance to the closest person.
10 int maxDistToClosest(vector<int>& seats) {
11 int firstOccupied = -1; // To mark the position of the first occupied seat.
12 int lastOccupied = -1; // To mark the position of the last occupied seat so far.
13 int maxDistance = 0; // To keep track of the maximum distance to the closest person so far.
14 int seatsCount = seats.size(); // Total number of seats.
15
16 // Iterate over the seats to find the maximum distance.
17 for (int i = 0; i < seatsCount; ++i) {
18 if (seats[i] == 1) { // When the seat is occupied
19 if (lastOccupied != -1) { // For all occupied seats except the first one
20 // Calculate the distance from the last occupied seat and update maxDistance if the distance is larger.
21 maxDistance = max(maxDistance, i - lastOccupied);
22 }
23 // Update the first occupied seat position upon finding the first occupied seat.
24 if (firstOccupied == -1) {
25 firstOccupied = i;
26 }
27 // Update the last occupied seat.
28 lastOccupied = i;
29 }
30 }
31
32 // Calculate the max distance for the edge seats and compare it with the previously found max.
33 // If an edge seat is the farthest from any occupied seat, update maxDistance accordingly.
34 // The distance from an edge to the first/last occupied seat is directly the index difference.
35 int edgeMaxDistance = max(firstOccupied, seatsCount - lastOccupied - 1);
36
37 // Return the maximum distance among the adjacent seats and the edge seats.
38 return max(maxDistance / 2, edgeMaxDistance);
39 }
40};
41
1/**
2 * Finds the maximum distance to the closest person in a row of seats.
3 * Seats are represented by an array where 1 is a person sitting and 0 is an empty seat.
4 * @param {number[]} seats - An array representing a row of seats.
5 * @returns {number} - The maximum distance to the closest person.
6 */
7function maxDistToClosest(seats: number[]): number {
8 let firstOccupiedSeatIndex = -1; // Index of the first occupied seat
9 let lastOccupiedSeatIndex = -1; // Index of the last occupied seat
10 let maxDistance = 0; // Maximum distance to the closest person
11 let totalSeats = seats.length; // Total number of seats
12
13 for (let i = 0; i < totalSeats; ++i) {
14 if (seats[i] === 1) {
15 if (lastOccupiedSeatIndex !== -1) {
16 // Update max distance calculating the half distance between two occupied seats
17 maxDistance = Math.max(maxDistance, i - lastOccupiedSeatIndex);
18 }
19 if (firstOccupiedSeatIndex === -1) {
20 // Update the index of the first occupied seat found
21 firstOccupiedSeatIndex = i;
22 }
23 // Update the index of the last occupied seat found
24 lastOccupiedSeatIndex = i;
25 }
26 }
27 // Calculate the max distance considering the edge seats and the max distance
28 // found between two occupied seats, where distance is halved since you sit in-between.
29 return Math.max(
30 firstOccupiedSeatIndex, // Distance from the first seat to the first person
31 totalSeats - lastOccupiedSeatIndex - 1, // Distance from the last person to the last seat
32 Math.floor(maxDistance / 2) // Max distance divided by 2 for the middle seat between two people
33 );
34}
35
Time and Space Complexity
Time Complexity
The time complexity of the given code is O(n)
, where n
is the length of the seats
list. This is derived from the fact that there is a single for loop that iterates over each element of the seats
list exactly once to find the maximum distance to the closest person.
Space Complexity
The space complexity of the code is O(1)
which means it uses constant extra space. This is because the variables first
, last
, and d
are used to keep track of positions and distance within the algorithm, regardless of the input size. The list itself is not copied or modified, and no additional data structures are utilized that grow with the input size.
Learn more about how to find time and space complexity quickly using problem constraints.
Which of the tree traversal order can be used to obtain elements in a binary search tree in sorted order?
Recommended Readings
LeetCode Patterns Your Personal Dijkstra's Algorithm to Landing Your Dream Job The goal of AlgoMonster is to help you get a job in the shortest amount of time possible in a data driven way We compiled datasets of tech interview problems and broke them down by patterns This way we
Recursion Recursion is one of the most important concepts in computer science Simply speaking recursion is the process of a function calling itself Using a real life analogy imagine a scenario where you invite your friends to lunch https algomonster s3 us east 2 amazonaws com recursion jpg You first
Runtime Overview When learning about algorithms and data structures you'll frequently encounter the term time complexity This concept is fundamental in computer science and offers insights into how long an algorithm takes to complete given a certain input size What is Time Complexity Time complexity represents the amount of time
Want a Structured Path to Master System Design Too? Don’t Miss This!