集册 Java实例教程 随机队列

随机队列

欢马劈雪     最近更新时间:2020-01-02 10:19:05

692
随机队列

import java.util.Arrays;

import java.util.Iterator;


public class RandomizedQueue<Item> implements Iterable<Item> {
/**
 from
* n o w j a v a . c o m 
**/

    private int numelements;

    private Object[] data;


    // construct an empty randomized queue

    public RandomizedQueue() {

        numelements = 0;

        data = new Object[1];

    }


    // is the queue empty?

    public boolean isEmpty() {

        return numelements == 0;/**来 自 N o w J a v a . c o m**/

    }


    // return the number of items on the queue

    public int size() {

        return numelements;

    }


    // resize array as needed

    private void resize(int newsize) {

        if (newsize < 1)

            return;

        Object[] newdata = new Object[newsize];

        for (int i = 0; i < numelements; i++) {

            newdata[i] = data[i];

            data[i] = null;

        }

        data = newdata;

    }


    // add the item

    public void enqueue(Item item) {

        if (item == null)

            throw new NullPointerException();

        if (numelements == data.length)

            resize(data.length * 2);

        data[numelements] = item;

        numelements += 1;

    }


    // delete and return a random item

    public Item dequeue() {

        if (numelements == 0)

            throw new java.util.NoSuchElementException();

        int idx = StdRandom.uniform(numelements);

        Item randval = (Item) data[idx];

        data[idx] = data[numelements - 1];

        data[numelements - 1] = null;

        numelements -= 1;

        if (numelements <= data.length / 4)

            resize(data.length / 2);

        return randval;

    }


    // return (but do not delete) a random item

    public Item sample() {

        if (numelements == 0)

            throw new java.util.NoSuchElementException();

        return (Item) data[StdRandom.uniform(numelements)];

    }


    // return an independent iterator over items in random order

    public Iterator<Item> iterator() {

        return new RandomizedQueueIterator();

    }


    private class RandomizedQueueIterator implements Iterator<Item> {

        private int current = 0;

        private int[] iterorder;


        public RandomizedQueueIterator() {

            if (numelements > 0) {

                iterorder = new int[numelements];

                for (int i = 0; i < numelements; i++)

                    iterorder[i] = i;

                StdRandom.shuffle(iterorder, 0, numelements - 1);

            }

        }


        public boolean hasNext() {

            return current < numelements;

        }


        public void remove() {

            throw new UnsupportedOperationException();

        }


        public Item next() {

            if (current >= numelements)

                throw new java.util.NoSuchElementException();

            Item item = (Item) data[iterorder[current]];

            current = current + 1;

            return item;

        }

    }


    // unit testing

    public static void main(String[] args) {

        RandomizedQueue<Integer> randqueue = new RandomizedQueue<Integer>();

        int N = Integer.parseInt(args[0]);

        double prob = Double.parseDouble(args[1]);

        // testing enqueue-dequeue

        System.out.println(
展开阅读全文