#1912

Design Movie Rental System

expert · 1145 · lc hard +32 · 62.5% accepted · 548 likes · top 64%

Description

You operate a movie rental service across n shops. Each entry entries[i] = [shopi, moviei, pricei] means shop shopi stocks movie moviei at pricei.

Implement MovieRentingSystem:

- MovieRentingSystem(int n, int[][] entries) — Initializes the system.

- List<Integer> search(int movie) — Returns up to 5 shops with the cheapest unrented copy of movie (sort by price asc, then shop ID asc).

- void rent(int shop, int movie) — Marks the copy rented.

- void drop(int shop, int movie) — Returns the copy.

- List<List<Integer>> report() — Returns up to 5 cheapest currently rented copies as [shop, movie] (sort by price asc, then shop ID asc, then movie ID asc).

Example 1:

Input

["MovieRentingSystem", "search", "rent", "rent", "report", "drop", "search"]

[[3, [[0, 1, 5], [0, 2, 6], [0, 3, 7], [1, 1, 4], [1, 2, 7], [2, 1, 5]]], [1], [0, 1], [1, 2], [], [1, 2], [2]]

Output

[null, [1, 0, 2], null, null, [[0, 1], [1, 2]], null, [0, 1]]

Example 2:

Explanation

MovieRentingSystem movieRentingSystem = new MovieRentingSystem(3, [[0, 1, 5], [0, 2, 6], [0, 3, 7], [1, 1, 4], [1, 2, 7], [2, 1, 5]]);

movieRentingSystem.search(1); // return [1, 0, 2], Movies of ID 1 are unrented at shops 1, 0, and 2. Shop 1 is cheapest; shop 0 and 2 are the same price, so order by shop number.

movieRentingSystem.rent(0, 1); // Rent movie 1 from shop 0. Unrented movies at shop 0 are now [2,3].

movieRentingSystem.rent(1, 2); // Rent movie 2 from shop 1. Unrented movies at shop 1 are now [1].

movieRentingSystem.report(); // return [[0, 1], [1, 2]]. Movie 1 from shop 0 is cheapest, followed by movie 2 from shop 1.

movieRentingSystem.drop(1, 2); // Drop off movie 2 at shop 1. Unrented movies at shop 1 are now [1,2].

movieRentingSystem.search(2); // return [0, 1]. Movies of ID 2 are unrented at shops 0 and 1. Shop 0 is cheapest, followed by shop 1.

Code

class MovieRentingSystem:

    def __init__(self, n: int, entries: List[List[int]]):
        

    def search(self, movie: int) -> List[int]:
        

    def rent(self, shop: int, movie: int) -> None:
        

    def drop(self, shop: int, movie: int) -> None:
        

    def report(self) -> List[List[int]]:
        


# Your MovieRentingSystem object will be instantiated and called as such:
# obj = MovieRentingSystem(n, entries)
# param_1 = obj.search(movie)
# obj.rent(shop,movie)
# obj.drop(shop,movie)
# param_4 = obj.report()

class MovieRentingSystem:

    def __init__(self, n: int, entries: List[List[int]]):
        

    def search(self, movie: int) -> List[int]:
        

    def rent(self, shop: int, movie: int) -> None:
        

    def drop(self, shop: int, movie: int) -> None:
        

    def report(self) -> List[List[int]]:
        


# Your MovieRentingSystem object will be instantiated and called as such:
# obj = MovieRentingSystem(n, entries)
# param_1 = obj.search(movie)
# obj.rent(shop,movie)
# obj.drop(shop,movie)
# param_4 = obj.report()