洗牌算法

Posted 2021-03-15 catlkb

洗牌算法

洗牌算法，刚在知乎这个回答上看到的一个算法，非常有趣。
通过概率论的知识原地实现了一个公平的随机算法。
大致的过程就是一个数组（假设有n个数），从后往前取第一个数A，第二个数随机从前面的数据中选取。最后算下来随机的概率是1/n。

代码如下

/**
 * @program: erp-user
 * @description:
 * @author: lkb
 * @create: 2020-01-16 13:22
 */
public class Test {

    public static void main(String[] args) {
        int[] arr = new int[]{6,34,27,4,9};
        System.out.println("初始数组：");
        Arrays.stream(arr).forEach(d-> System.out.print(d +", "));
        System.out.println();
        for(int i=arr.length-1;i>0;i--){
            int j = new Random().nextInt(i);
            swap(arr,i,j);
            System.out.println(arr[i] +" <-> " + arr[j] );
            Arrays.stream(arr).forEach(d-> System.out.print(d +", "));
            System.out.println();
        }
    }


    private static void swap(int[] arr, int a,int b){
        int tmp = arr[a];
        arr[a] = arr[b];
        arr[b] = tmp;
    }
}