算法笔记六、希尔排序代码

package com.hao.firstdemo.datastruct.sort;

import java.util.Arrays;

/**
 * @author haoxiansheng
 * @data 2020/5/7 23:00
 */
public class ShellSort {
    public static void main(String[] args) {
        int[] arr = {8, 9, 1, 7, 2, 3, 5, 4, 6, 0};
        // 测试
        shellSort(arr);
        shellSort2(arr);
    }

    /**
     * 希尔排序交换法  这个时间比插入排序慢了
     *
     * @param arr
     * @return
     */
    public static int[] shellSort(int[] arr) {
        int temp = 0;
        int count = 1;
        // 分组
        for (int gap = arr.length / 2; gap > 0; gap /= 2) {
            for (int i = gap; i < arr.length; i++) {

                // 遍历各组中的元素 每组（gap个）
                for (int j = i - gap; j >= 0; j -= gap) {
                    if (arr[j] > arr[j + gap]) { //当前元素大于后面那个元素
                        temp = arr[j];
                        arr[j] = arr[j + gap];
                        arr[j + gap] = temp;
                    }
                }
            }
            System.out.printf("希尔排序交换第%d轮 为=》%s", count, Arrays.toString(arr));
            System.out.println();
            count++;
        }
        return arr;
    }

    /**
     * 希尔排序移位法
     *
     * @param arr
     * @return
     */
    public static int[] shellSort2(int[] arr) {

        for (int gap = arr.length / 2; gap > 0; gap /= 2) {
            int temp = 0;
            int count = 1;
            for (int i = gap; i < arr.length; i++) {
                int j = i;
                temp = arr[i];
                if (arr[j] < arr[j - gap]) {
                    while (j - gap >= 0 && temp < arr[j - gap]) {
                        // 移动
                        arr[j] = arr[j - gap];
                        j -= gap;
                    }
                    // 当退出循环,就给temp找到插入位置 这可以优化
                    arr[j] = temp;
                }
            }
            System.out.printf("希尔排序位移第%d轮 为=》%s", count, Arrays.toString(arr));
            System.out.println();
            count++;
        }
        return arr;
    }

}