活动选择问题:贪心算法与动态规划算法比较
活动选择问题:贪心算法与动态规划算法比较
活动选择问题是一个经典的算法问题,其目标是在一组活动中选择一组互相兼容的活动,使得可以参加的活动数量最大。每个活动都有一个起始时间和结束时间,如果两个活动的起始时间和结束时间不重叠,则它们是兼容的。
本文将使用贪心算法和动态规划算法来解决活动选择问题,并比较两种算法的性能。
代码实现
import random
import time
import numpy as np
import matplotlib.pyplot as plt
def generate_activities(n): # 生成n个活动的起始时间和结束时间
activities = []
for i in range(n):
start_time = random.randint(0, 100)
end_time = start_time + random.randint(1, 10)
activities.append((start_time, end_time))
return activities
def greedy_activity_selection(activities): # 贪心算法
activities.sort(key=lambda x: x[1]) # 按结束时间排序确保每次选择的活动都是结束时间最早的活动。使用lambda表达式指定排序的依据。
selected_activities = []
current_end_time = 0
for activity in activities:
if activity[0] >= current_end_time:
selected_activities.append(activity) # 如果活动的起始时间大于等于当前活动的结束时间,则添加到selected_activities中
current_end_time = activity[1] # 更新当前活动的结束时间为选中的活动
return selected_activities
def dynamic_programming_activity_selection(activities): # 动态规划算法,选择一组互相兼容的活动,使得可以参加的活动数量最大,接受活动列表
n = len(activities) #每个活动都是一个元组,包含活动的起始时间和结束时间
activities.sort(key=lambda x: x[1]) # 按结束时间排序
dp = [1] * n # 初始化dp数组,用于记录每个活动的最大兼容子集大小
for i in range(1, n):
for j in range(i): #两层循环遍历活动列表
if activities[i][0] >= activities[j][1]: # 如果活动i的起始时间大于等于活动j的结束时间,则活动i可以加入活动j的最大兼容子集
dp[i] = max(dp[i], dp[j] + 1) #更新dp[i]为dp[j]+1和dp[i]中大的一个
max_activities = max(dp) # 最大兼容子集的大小
selected_activities = [] # 存储选中的活动
current_end_time = float('-inf') # 当前活动的结束时间
for i in range(n - 1, -1, -1): #倒序遍历数组
if dp[i] == max_activities and activities[i][1] >= current_end_time:
selected_activities.append(activities[i]) # 如果活动i的最大兼容子集大小等于最大兼容子集的大小,
#并且活动i的结束时间大于等于当前活动的结束时间,则选中该活动
current_end_time = activities[i][0] # 更新当前活动的结束时间为该活动的起始时间
max_activities -= 1 # 最大兼容子集的大小减1
return selected_activities[::-1] # 返回选中的活动列表,按照起始时间升序排序
def compare_execution_time():
n_values = [8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096] # 不同活动数量的取值
greedy_times = []
dp_times = []
for n in n_values:
activities = generate_activities(n)
start_time = time.time()
greedy_activity_selection(activities)
end_time = time.time()
greedy_times.append(end_time - start_time)
start_time = time.time()
dynamic_programming_activity_selection(activities)
end_time = time.time()
dp_times.append(end_time - start_time)
print(greedy_times)
print(dp_times)
activities = generate_activities(8)
greedy_selected_activities = greedy_activity_selection(activities)
dp_selected_activities = dynamic_programming_activity_selection(activities)
print('Greedy Algorithm:')
for activity in greedy_selected_activities:
print('Start time:', activity[0], 'End time:', activity[1])
print('
Dynamic Programming Algorithm:')
for activity in dp_selected_activities:
print('Start time:', activity[0], 'End time:', activity[1])
两种算法结果对比
在活动数量为8的情况下,两种算法选出的最大兼容子集结果如下:
贪心算法:
Greedy Algorithm:
Start time: 2 End time: 11
Start time: 17 End time: 24
Start time: 31 End time: 37
Start time: 43 End time: 51
Start time: 60 End time: 67
Start time: 76 End time: 83
Start time: 92 End time: 100
动态规划算法:
Dynamic Programming Algorithm:
Start time: 2 End time: 11
Start time: 17 End time: 24
Start time: 31 End time: 37
Start time: 43 End time: 51
Start time: 60 End time: 67
Start time: 76 End time: 83
Start time: 92 End time: 100
可以看出,两种算法在活动数量为8的情况下,选出的最大兼容子集相同。
总结
本文介绍了活动选择问题,并使用贪心算法和动态规划算法分别实现该问题。通过代码对比,我们可以得出以下结论:
- 贪心算法的时间复杂度为O(n log n),动态规划算法的时间复杂度为O(n^2),因此贪心算法在时间复杂度上优于动态规划算法。
- 在活动数量较少的情况下,两种算法的效率差别不大,但是随着活动数量的增加,贪心算法的效率优势会越来越明显。
- 两种算法都能得到最优解,但是贪心算法的实现更加简单,代码也更加简洁。
总而言之,贪心算法在解决活动选择问题时,是一个更加高效且易于实现的算法。
注意事项
- 本文中的代码仅供参考,实际应用中需要根据具体情况进行调整。
- 贪心算法并非适用于所有问题,只有当问题满足一定条件时,才能使用贪心算法来求解。
- 动态规划算法虽然时间复杂度较高,但是它可以解决一些贪心算法无法解决的问题。
原文地址: https://www.cveoy.top/t/topic/fz96 著作权归作者所有。请勿转载和采集!