用Scala瞎写的直接插入排序算法

继续用Scala瞎写，这次写的是直接插入排序算法。（插入算法有很多种，直接插入算法是其中一种，其它的还有二分插入，链表插入等）

 

直接插入排序的基本概念

直接插入排序的概念是通过构建有序的序列，对于未排序的数据，在已排序的序列中从后向前扫描，找到相应位置并插入。

 

拿数组来说

首先比较数组的头两个数，将它们变为有序的，接下来将数组的第三个数与前两个数从后向前扫描，把第三个数插入到有序的序列中，再处理第四个数，一直到完。

 

把它写出来

根据上面的描述，我决定用for循环来遍历整个数组，每次从数组中取一个数，然后将这个数和它前一个数进行比较，如果这个数大于它前面的那个数，因为前面已经是有序的序列了，所以这个数可以已经处于有序的位置上了；如果这个数小于它前面的那个数，那么就把它俩互换，继续向前寻找这个数的位置，一直找到比它小的那个数后面。

def sort(arr:Array[Int])={ for((index, item)<-(0 until arr.length) zip arr){ if(index!=0 && item<arr(index-1)){ var insertIndex=index-1 while(insertIndex>=0){ if(item<=arr(insertIndex)){ arr(insertIndex+1)=arr(insertIndex) arr(insertIndex)=item } insertIndex-=1 } } } }

 

完善一下

用它来处理整数数组应该是没有问题了。但是有两个地方让我感觉不舒服：第一就是排序应该不仅仅局限于整数，还有Long，Double甚至是Char也能排序；第二就是插入排序并没有指出应该是从小到大还是从大到小排列，所以这个应该让调用的人自己决定。

对于第一个问题，简单的引入Scala的View Bound就可以了；至于第二个，就需要一个比较函数传进来了，还好Scala的FP功能可以让我简单的实现这一点：

def sort[T <% Ordered[T]](arr:Array[T])(f:(T,T)=>Boolean)={ for((index, item)<-(0 until arr.length) zip arr){ if(index!=0 && f(item,arr(index-1))){ var insertIndex=index-1 while(insertIndex>=0){ if(f(item,arr(insertIndex))){ arr(insertIndex+1)=arr(insertIndex) arr(insertIndex)=item } insertIndex-=1 } } } }

 

算法复杂度

通过代码能够看出来，直接插入排序空间是恒定的，但是就从它的循环套循环来看，如果是最坏的情况，需要进行n(n-1)/2比较，平均来看，时间复杂度是O(n² )。看来这个对付小的数组排序还是可以的，如果是大的数据，还是算了吧。

 

测试代码

object InsertionSortMain { def main(args: Array[String]): Unit = { val array = Array(2,1,4,6,2,9,66,54,78,91,20,47,0,4,-23) sort(array){_>_} array foreach println } def sort[T <% Ordered[T]](arr:Array[T])(f:(T,T)=>Boolean)={ for((index, item)<-(0 until arr.length) zip arr){ if(index!=0 && f(item,arr(index-1))){ var insertIndex=index-1 while(insertIndex>=0){ if(f(item,arr(insertIndex))){ arr(insertIndex+1)=arr(insertIndex) arr(insertIndex)=item } insertIndex-=1 } } } } }