heap sort

cschen1205 · cschen1205 · commit 461eb07aa329 · 2017-05-25T13:25:44.000+08:00
diff --git a/src/jssort.js b/src/jssort.js
@@ -217,7 +217,44 @@ var jssort = jssort || {};
         jss.threeWaysQuickSort(a, lo, lt-1, compare);
         jss.threeWaysQuickSort(a, gt+1, hi, compare);
     };
+    
+    jss.heapSort = function(a, compare) {
+        if (!compare) {
+            compare = function (a1, a2){
+                return a1 - a2;
+            };
+        }
+        
+        var N = a.length;
+        var N2 = Math.floor(N / 2);
+        for (var k = N2; k <= 1; --k){
+            jss.sink(a, k, N);
+        }
+        
+        while (N > 1) {
+            jss.exchange(a, jss.heapIndex(1), jss.heapIndex(N));
+            N--;
+            jss.sink(a, 1, N);
+        }
+    };
+    
+    jss.sink = function(a, k, N) {
+        while (k * 2 <= N) {
+            var child =  k * 2;
+            if(child < N && jss.less(a[jss.heapIndex(child), jss.heapIndex(child+1)])) {
+                child++;
+            }
+            if(jss.less(a[jss.heapIndex(k)], a[jss.heapIndex(child)])) {
+                jss.exchange(a, jss.heapIndex(k), jss.heapIndex(child));
+            } else {
+                break;
+            }
+        }  
+    };
 
+    jss.heapIndex = function(i) {
+        return i - 1;
+    };
 })(jssort);
 
 if(module) {
diff --git a/test/three-ways-quick-sort-spec.js b/test/three-ways-quick-sort-spec.js
@@ -8,7 +8,6 @@ describe("Three Ways Quick Sort", function() {
     	
         var a = [3, 4, 5, 1, 2, 4, 6, 8, 9, 3, 4, 67, 34, 53, 44, 2];
         jssort.threeWaysQuickSort(a);
-        console.log(a);
         for(var i = 1; i < a.length; ++i){
             expect(a[i-1]).not.to.above(a[i]);
         }

Original file line number	Diff line number	Diff line change
`@@ -8,7 +8,6 @@ describe("Three Ways Quick Sort", function() {`
`8`	`8`
`9`	`9`	`var a = [3, 4, 5, 1, 2, 4, 6, 8, 9, 3, 4, 67, 34, 53, 44, 2];`
`10`	`10`	`jssort.threeWaysQuickSort(a);`
`11`		`- console.log(a);`
`12`	`11`	`for(var i = 1; i < a.length; ++i){`
`13`	`12`	`expect(a[i-1]).not.to.above(a[i]);`
`14`	`13`	`}`