Python动画可视化排序算法:冒泡排序,插入排序,选择排序,堆排序,shell排序,合并排序,快速排序。
冒泡排序
冒泡排序是最直接的排序算法之一,它会多次遍历一个列表。
- 从第一个元素开始,将当前元素与数组的下一个元素进行比较。
- 如果当前元素大于数组的下一个元素,则交换它们。
- 如果当前元素小于下一个元素,则移动到下一个元素。
从本质上讲,每个项目都会“冒泡”到它所属的位置。
import random
import math
from array import *
from math import pi
import numpy as np
############################################################
# Bubble Sort Algorithm
############################################################
def bubble_sort(arr, count):
for i in range(count):
#insert keyframe for every cube on every frame
for cube in arr:
cube.keyframe_insert(data_path="location", frame=i)
already_sorted = True
for j in range(count - i -1):
#get materials
mat1 = arr[j].active_material.diffuse_color
mat2 = arr[j + 1].active_material.diffuse_color
rg1, rg2 = get_rg(mat1, mat2)
#compare first colorarray values
if rg1 > rg2:
#change location & insert keyframes based on bubble sort
arr[j].location.x = (j+1)*2
arr[j].keyframe_insert(data_path="location", frame=i+1)
arr[j+1].location.x = j*2
arr[j+1].keyframe_insert(data_path="location", frame=i+1)
#rearrange arrays
arr[j], arr[j + 1] = arr[j + 1], arr[j]
already_sorted = False
if already_sorted:
break
############################################################
# Setup Random Colors + Array to be sorted
############################################################
############################################################
# Get R and G Values from Material
############################################################
def get_rg(mat1, mat2):
#get R value of both materials
r1 = mat1[0]
r2 = mat2[0]
#get G value of both materials
g1 = mat1[1]
g2 = mat2[1]
# R + G = value for comparison
rg1 = r1 + g1
rg2 = r2 + g2
return rg1, rg2
############################################################
# Call Functions
############################################################
#setup_array(number of planes)
Matrix, count = setup_array(24)#only even numbers are valid
#bubble_sort + visualisation
for i in range(count):
bubble_sort(Matrix[i], count)
插入排序
与冒泡排序一样,插入排序算法易于实现和理解。
- 在数组上从 arr[1] 迭代到 arr[n]。
- 将当前元素(键)与其前身进行比较。
- 如果关键元素小于其前任元素,则比较之前的元素。
- 将较大的元素向上移动一位,为交换的元素腾出空间。
它将给定的数组拆分为已排序和未排序的部分,然后将未排序部分的值拾取并放置在已排序部分的正确位置。