码迷,mamicode.com
首页 > 编程语言 > 详细

【算法题】04-猫狗队列

时间:2020-03-06 15:33:38      阅读:66      评论:0      收藏:0      [点我收藏+]

标签:exception   宠物   class   com   imp   根据   计数器   思路   调用   

题目

宠物猫和狗的类如下:

public static class Pet {
        private String type;

        public Pet(String type) {
            this.type = type;
        }

        public String getPetType() {
            return this.type;
        }
    }

    public static class Dog extends Pet {
        public Dog() {
            super("dog");
        }
    }

    public static class Cat extends Pet {
        public Cat() {
            super("cat");
        }
    }

实现个猫狗队列的结构,需求如下:

  • 用户可以调用add方法将cat类或dog类的实例放入队列中
  • 用户可以调用pullAll方法,将队列中所有的实例安装进队列的先后顺序依次弹出
  • 用户可以调用pollDog方法,将队列中的dog类的实例按照进队列的先后顺序依次弹出
  • 用户可以调用pollCat方法,将队列中的cat类的实例按照进队列的先后顺序依次弹出
  • 用户可以调用isDogEmpty方法,检查队列中是否有dog类的实例
  • 用户可以调用isCatEmpty方法,检查队列中是否有cat类的实例

思路

将不同的实例盖上时间戳的方法,但是又不能改变用户本身的类,所以定义一个新的类

/**
     * @Desc 新增一个类  为每个对象加上一个时间戳计数
     * @author sangliping
     *
     */
    public static class PetEnterQueue {
        private Pet pet;
        private long count;

        public PetEnterQueue(Pet pet, long count) {
            this.pet = pet;
            this.count = count;
        }

        public Pet getPet() {
            return this.pet;
        }

        public long getCount() {
            return this.count;
        }

        public String getEnterPetType() {
            return this.pet.getPetType();
        }
    }

我们实现的队列其实就是PetEnterQueue类的实例,大体来说,首先要有一个不断累加的数据项,用来表示实例进队的时间,同时有两个队列,一个是只放dog类实例的队列dogQ另一个是只放cat类的实例的队列catQ.在加入实例时,

加入实例时,如果是dog,就盖上时间戳,生成对应的PetEnterQueue类的实例;然后放入dogQ;如果实例是cat,就盖上时间戳,生成对应的PetEnterQueue类的实例,然后放入catQ.

只想弹出dog类的实例时,从dogQ里不断弹出即可,弹出cat同理.

按实际顺序弹出时,因为dogQ的队列头表示所有dog实例中最早进队额的实例,catQ同理,这样的话比较队头元素的时间戳,谁早就弹出谁。

解题

package com.zixin.learn;

import java.util.LinkedList;
import java.util.Queue;

public class DogCatQueue {

    public static class Pet {
        private String type;

        public Pet(String type) {
            this.type = type;
        }

        public String getPetType() {
            return this.type;
        }
    }

    public static class Dog extends Pet {
        public Dog() {
            super("dog");
        }
    }

    public static class Cat extends Pet {
        public Cat() {
            super("cat");
        }
    }

    /**
     * @Desc 新增一个类  为每个对象加上一个时间戳计数
     * @author sangliping
     *
     */
    public static class PetEnterQueue {
        private Pet pet;
        private long count;

        public PetEnterQueue(Pet pet, long count) {
            this.pet = pet;
            this.count = count;
        }

        public Pet getPet() {
            return this.pet;
        }

        public long getCount() {
            return this.count;
        }

        public String getEnterPetType() {
            return this.pet.getPetType();
        }
    }

    public static class DogCatQueue {
        //dog队列
        private Queue<PetEnterQueue> dogQ;
        //cat队列
        private Queue<PetEnterQueue> catQ;
        //计数器
        private long count;

        //构造函数,初始化一个catQ和dogQ 并设置计数器为0
        public DogCatQueue() {
            this.dogQ = new LinkedList<PetEnterQueue>();
            this.catQ = new LinkedList<PetEnterQueue>();
            this.count = 0;
        }

        /**
         * 队列中添加元素,根据类型来判断进入那个队列
         * @param pet
         */
        public void add(Pet pet) {
            if (pet.getPetType().equals("dog")) {
                this.dogQ.add(new PetEnterQueue(pet, this.count++));
            } else if (pet.getPetType().equals("cat")) {
                this.catQ.add(new PetEnterQueue(pet, this.count++));
            } else {
                throw new RuntimeException("err, not dog or cat");
            }
        }

        /**
           * 按进队的顺序弹出一个元素 两个队列都不为空的时候,比较队头哪个小弹出哪个
         * @return
         */
        public Pet pollAll() {
            if (!this.dogQ.isEmpty() && !this.catQ.isEmpty()) {
                //peek()获取元素  不弹出
                if (this.dogQ.peek().getCount() < this.catQ.peek().getCount()) {
                    //dog的计数小弹dog
                    return this.dogQ.poll().getPet();
                } else {
                    //cat的计数小弹cat
                    return this.catQ.poll().getPet();
                }
                //dog不为空,弹出dog
            } else if (!this.dogQ.isEmpty()) {
                return this.dogQ.poll().getPet();
                //cat不为空,弹出cat
            } else if (!this.catQ.isEmpty()) {
                return this.catQ.poll().getPet();
            } else {
                throw new RuntimeException("err, queue is empty!");
            }
        }

        /**
         * @Desc dog不为空,则弹出dog
         * @return
         */
        public Dog pollDog() {
            if (!this.isDogQueueEmpty()) {
                return (Dog) this.dogQ.poll().getPet();
            } else {
                throw new RuntimeException("Dog queue is empty!");
            }
        }
        /**
         * @Desc cat不为空,则弹出cat
         * @return
         */
        public Cat pollCat() {
            if (!this.isCatQueueEmpty()) {
                return (Cat) this.catQ.poll().getPet();
            } else
                throw new RuntimeException("Cat queue is empty!");
        }

        public boolean isEmpty() {
            return this.dogQ.isEmpty() && this.catQ.isEmpty();
        }

        public boolean isDogQueueEmpty() {
            return this.dogQ.isEmpty();
        }

        public boolean isCatQueueEmpty() {
            return this.catQ.isEmpty();
        }

    }

    public static void main(String[] args) {
        DogCatQueue test = new DogCatQueue();

        Pet dog1 = new Dog();
        Pet cat1 = new Cat();
        Pet dog2 = new Dog();
        Pet cat2 = new Cat();
        Pet dog3 = new Dog();
        Pet cat3 = new Cat();

        test.add(dog1);
        test.add(cat1);
        test.add(dog2);
        test.add(cat2);
        test.add(dog3);
        test.add(cat3);

        test.add(dog1);
        test.add(cat1);
        test.add(dog2);
        test.add(cat2);
        test.add(dog3);
        test.add(cat3);

        test.add(dog1);
        test.add(cat1);
        test.add(dog2);
        test.add(cat2);
        test.add(dog3);
        test.add(cat3);
        while (!test.isDogQueueEmpty()) {
            System.out.println(test.pollDog().getPetType());
        }
        while (!test.isEmpty()) {
            System.out.println(test.pollAll().getPetType());
        }
    }

}

【算法题】04-猫狗队列

标签:exception   宠物   class   com   imp   根据   计数器   思路   调用   

原文地址:https://www.cnblogs.com/dream-to-pku/p/12426233.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!