快速排序算法在处理不同容量数组时的数据可视化

回忆快速排序:

链接: link

javascript 复制代码
#将不同数据规模数组快排时间可视化
import time
import random
import matplotlib.pyplot as plt
import numpy as np
#三值取中法取轴值
def FindPivox(nums,left,right):
     mid=(left+right)//2
     if nums[left]>nums[mid]:
         nums[left],nums[mid]=nums[mid],nums[left]
     if nums[left]>nums[right]:
         nums[left],nums[right]=nums[right],nums[left]
     if nums[mid]>nums[right]:
         nums[mid],nums[right]=nums[right],nums[mid]
     return mid

def partition(nums,left,right,pivox):
    while left<=right:
        while nums[left]<pivox:
            left+=1
        while nums[right]>pivox and right>0:
            right-=1
        nums[left],nums[right]=nums[right],nums[left]
    nums[left], nums[right] = nums[right], nums[left]
    return left

def quick_sort(nums,left=0,right=None):
    if right is None:
        right=len(nums)-1
    #1、确定恰当的轴值
    p_index1=FindPivox(nums,left,right)
    #2、轴值和右端元素交换位置,方便双指针法分割小值序列和大值序列
    nums[p_index1],nums[right]=nums[right],nums[p_index1]
    #3、双指针法分割小值序列和大值序列
    p_index2=partition(nums,left,right-1,nums[right])
    #4、将轴值元素放在小值元素序列右边,大值元素左边,其实就是左指针元素和轴值交换
    nums[p_index2],nums[right]=nums[right],nums[p_index2]
    #5、递归:(增加条件当数据规模小于时停止递归改用直接插入排序)
    if (p_index2-left)>1:#隐含了基准情形,当序列中元素为1时排序完成
        quick_sort(nums,left,p_index2-1)
    if (right-p_index2)>1:
        quick_sort(nums,p_index2+1,right)
        return nums

#数据可视化

# 生成的规模为volume的整数数组
def generateData(volume,repetitionRate):
    # :param volume:定义生成的数据规模大小,应为整数
    # :param repetitionRate: 需要生成的数据序列的重复率,其值为[0,1]的小数,0代表完全不重复,1代表全部重复
    # :return: 按照要求生成的规模为volume的整数数组
    dataSet = volume - int(repetitionRate*volume)
    if dataSet == 0:
        return [0]*volume
    data = [0]*volume
    for i in range(dataSet):
        data[i] = i
    startIndex = dataSet
    while startIndex < volume:
        for i in range(dataSet):
            data[startIndex] = random.randrange(0,dataSet)
            startIndex += 1
            if startIndex == volume:
                 break
    random.shuffle(data) # 将数据随机打乱
    return data

#运行时间
def elapsedTime(function, volumes):
    # :param function:需要计算运行函数的函数名
    # :param volumes: 记录数据规模大小的数组
    # :return: 记录运行时间的数组
    elapsedTimes = []
    for volume in volumes:
        nums = generateData(volume, 0)
        startTime = time.time_ns()
        function(nums,0,len(nums)-1)
        endTime = time.time_ns()
        elapsedTimes.append(endTime-startTime)
    return elapsedTimes

#可视化,横轴竖轴分别代表,数据规模、运行时间
def getChart(X,Y):
    fig, ax = plt.subplots()#创建一个新的图形窗口和一个坐标轴对象
    ax.plot(X, Y)
    plt.xlabel('Volume')  #指定x轴名称
    plt.ylabel('Elapsed Time')  #指定y轴名称
    plt.yticks(np.arange(0,0.5,0.1))#设置y轴间距
    plt.title('Quick Sort Elapsed Time')  #创建图表名称
    plt.show()
    plt.pause(5)
def main():
    volumes = [10*i for i in range(1,3)]#类型:列表。
    Y = elapsedTime(quick_sort,volumes)
    getChart(volumes, Y)#x:数据规模 y:快速排序所需时间

if __name__ == "__main__":
    main()

输出结果:

要想得到处理大规模数组所需的时间,可以修改

volumes参数

eg

javascript 复制代码
volumes = [100*i for i in range(1,101)]

这样就能得到快排在处理数据规模从100、200...10000的数组所需的时间啦

相关推荐
补三补四3 分钟前
走进量化投资【前章】
算法·机器学习·金融
Paddi9301 小时前
Codeforces Round 1004 (Div. 1) C. Bitwise Slides
c++·算法
Luis Li 的猫猫2 小时前
深度学习中的知识蒸馏
人工智能·经验分享·深度学习·学习·算法
查理零世2 小时前
【蓝桥杯集训·每日一题2025】 AcWing 6118. 蛋糕游戏 python
python·算法·蓝桥杯
敲代码的小王!2 小时前
MD5加密算法和BCrypt密码加密算法
java·算法·安全
带娃的IT创业者3 小时前
机器学习实战(6):支持向量机(SVM)——强大的非线性分类器
算法·机器学习·支持向量机
Java能学吗6 小时前
2.17学习总结
数据结构·学习
孑么6 小时前
力扣 买卖股票的最佳时机
算法·leetcode·职场和发展·贪心算法·动态规划
诚信爱国敬业友善7 小时前
常见排序方法的总结归类
开发语言·python·算法
gentle_ice7 小时前
跳跃游戏 II - 贪心算法解法
数据结构·算法·leetcode·贪心算法