leetcode-384-打乱数组(shuffle an array)-java

题目及测试

package pid384;
/* Shuffle an Array

打乱一个没有重复元素的数组。

示例:

// 以数字集合 1, 2 和 3 初始化数组。
int[] nums = {1,2,3};
Solution solution = new Solution(nums);

// 打乱数组 [1,2,3] 并返回结果。任何 [1,2,3]的排列返回的概率应该相同。
solution.shuffle();

// 重设数组到它的初始状态[1,2,3]。
solution.reset();

// 随机返回数组[1,2,3]打乱后的结果。
solution.shuffle();



*/




public class main {
	
	public static void main(String[] args) {
		int[][] testTable = {{1,2,3},{2,7,9,3,1},{7,10,4,3,1},{11,6,2,7}};
		for (int[] ito : testTable) {
			test(ito);
		}
	}
		 
	private static void test(int[] ito) {
		Solution solution = new Solution(ito);
		int[] rtn;
		long begin = System.currentTimeMillis();
		for (int i = 0; i < ito.length; i++) {
		    System.out.print(ito[i]+" ");		    
		}
		System.out.println();
		//开始时打印数组
		
		rtn = solution.shuffle();//执行程序
		long end = System.currentTimeMillis();	
		
		System.out.println(ito + ": rtn=" );
		System.out.println( " rtn=" );
		for (int i = 0; i < rtn.length; i++) {
		    System.out.print(rtn[i]+" ");
		}//打印结果几数组
		
		rtn = solution.reset();//执行程序
		
		System.out.println(ito + ": rtn=" );
		System.out.println( " rtn=" );
		for (int i = 0; i < rtn.length; i++) {
		    System.out.print(rtn[i]+" ");
		}//打印结果几数组
		
		System.out.println();
		System.out.println("耗时：" + (end - begin) + "ms");
		System.out.println("-------------------");
	}

}

解法1（成功，282ms，较慢）
在class中设置一个int数组作为源头
reset方法，返回一个copy的数组
shuffle方法，将源头数组一个个加入linkedlist
int index=(int)(Math.random()*i);
每次将list的这个index的数字取出加入结果数组
然后i–

package pid384;

import java.lang.reflect.Array;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;

class Solution {

	int[] source;
	
    public Solution(int[] nums) {
        source=nums;
    }
    
    /** Resets the array to its original configuration and return it. */
    public int[] reset() {
        int[] result=Arrays.copyOf(source,source.length);    	
    	return result;
    }
    
    /** Returns a random shuffling of the array. */
    public int[] shuffle() {
        List<Integer> list=new LinkedList<>();
        int length=source.length;
        if(length==0){
        	return null;
        }
        int[] result=new int[length];
        
        for(int i=0;i<length;i++){
        	list.add(source[i]);
        }
        for(int i=length;i>0;i--){
        	int index=(int)(Math.random()*i);
        	result[length-i]=list.get(index);
        	list.remove(index);
        }
        
        
    	return result;
    }
}

/**
 * Your Solution object will be instantiated and called as such:
 * Solution obj = new Solution(nums);
 * int[] param_1 = obj.reset();
 * int[] param_2 = obj.shuffle();
 */

解法2（成功，305s，较慢）
速度为O(n)
复制到一个新数组，根据交换，进行随机
Fisher-Yates 洗牌算法跟暴力算法很像。在每次迭代中，生成一个范围在当前下标到数组末尾元素下标之间的随机整数。接下来，将当前元素和随机选出的下标所指的元素互相交换 - 这一步模拟了每次从 “帽子” 里面摸一个元素的过程，其中选取下标范围的依据在于每个被摸出的元素都不可能再被摸出来了。此外还有一个需要注意的细节，当前元素是可以和它本身互相交换的 - 否则生成最后的排列组合的概率就不对了。

class Solution {

	int[] value;
	
    public Solution(int[] nums) {
        value=nums;
    }
    
    /** Resets the array to its original configuration and return it. */
    public int[] reset() {       
    	return value;
    }
    
    /** Returns a random shuffling of the array. */
    public int[] shuffle() {
        int length=value.length;
        if(length==0||length==1){
        	return value;
        }
        int[] result=new int[length];
        System.arraycopy(value, 0, result, 0, length);
        
        for(int i=length-1;i>=0;i--){
        	int index=(int)Math.floor(Math.random()*(i+1));
        	int temp=result[index];
        	result[index]=result[i];
        	result[i]=temp;
        }
    	
    	return result;
    }
}

解法3（别人的）
与解法2很像，但是打乱时不用复制一个新数组，代价是空间增加一个数组

class Solution {
    private int[] array;
    private int[] original;

    Random rand = new Random();

    private int randRange(int min, int max) {
        return rand.nextInt(max - min) + min;
    }

    private void swapAt(int i, int j) {
        int temp = array[i];
        array[i] = array[j];
        array[j] = temp;
    }

    public Solution(int[] nums) {
        array = nums;
        original = nums.clone();
    }
    
    public int[] reset() {
        array = original;
        original = original.clone();
        return original;
    }
    
    public int[] shuffle() {
        for (int i = 0; i < array.length; i++) {
            swapAt(i, randRange(i, array.length));
        }
        return array;
    }
}