diff --git a/DIRECTORY.md b/DIRECTORY.md index f0a34a553..c7dfdb674 100644 --- a/DIRECTORY.md +++ b/DIRECTORY.md @@ -526,6 +526,7 @@ * [Test Min Spanning Tree Prim](graphs/tests/test_min_spanning_tree_prim.py) ## Greedy Methods + * [Activity Selection](greedy_methods/activity_selection.py) * [Best Time To Buy And Sell Stock](greedy_methods/best_time_to_buy_and_sell_stock.py) * [Fractional Cover Problem](greedy_methods/fractional_cover_problem.py) * [Fractional Knapsack](greedy_methods/fractional_knapsack.py) diff --git a/greedy_methods/activity_selection.py b/greedy_methods/activity_selection.py new file mode 100644 index 000000000..6cdeff3d0 --- /dev/null +++ b/greedy_methods/activity_selection.py @@ -0,0 +1,52 @@ +""" +The Activity Selection Problem is a classic problem in which a set of activities, +each with a start and end time, needs to be scheduled in such a way that +the maximum number of non-overlapping activities is selected. +This is a greedy algorithm where at each step, +we choose the activity that finishes the earliest +and does not conflict with previously selected activities. + +Wikipedia: https://en.wikipedia.org/wiki/Activity_selection_problem +""" + + +def activity_selection(activities: list[tuple[int, int]]) -> list[tuple[int, int]]: + """ + Solve the Activity Selection Problem using a greedy algorithm by selecting + the maximum number of non-overlapping activities from a list of activities. + + Parameters: + activities: A list of tuples where each tuple contains + the start and end times of an activity. + + Returns: + A list of selected activities that are non-overlapping. + + Example: + >>> activity_selection([(1, 3), (2, 5), (3, 9), (6, 8)]) + [(1, 3), (6, 8)] + + >>> activity_selection([(0, 6), (1, 4), (3, 5), (5, 7), (5, 9), (8, 9)]) + [(1, 4), (5, 7), (8, 9)] + + >>> activity_selection([(1, 2), (2, 4), (3, 5), (0, 6)]) + [(1, 2), (2, 4)] + + >>> activity_selection([(5, 9), (1, 2), (3, 4), (0, 6)]) + [(1, 2), (3, 4), (5, 9)] + """ + + # Step 1: Sort the activities by their end time + sorted_activities = sorted(activities, key=lambda activity: activity[1]) + + # Step 2: Select the first activity (the one that finishes the earliest) + # as the initial activity + selected_activities = [sorted_activities[0]] + + # Step 3: Iterate through the sorted activities and select the ones + # that do not overlap with the last selected activity + for i in range(1, len(sorted_activities)): + if sorted_activities[i][0] >= selected_activities[-1][1]: + selected_activities.append(sorted_activities[i]) + + return selected_activities