Go-实现经典的猫狗队列

怼烎@ 2022-09-09 13:00 172阅读 0赞

代码片段如下：

/*
     *                    .::::.
     *                  .::::::::.
     *                 :::::::::::
     *             ..:::::::::::'
     *           '::::::::::::'
     *             .::::::::::
     *        '::::::::::::::..
     *             ..::::::::::::.		Utils:狗猫队列
     *           ``::::::::::::::::		Author:崔金朋
     *            ::::``:::::::::'        .:::.
     *           ::::'   ':::::'       .::::::::.
     *         .::::'      ::::     .:::::::'::::.
     *        .:::'       :::::  .:::::::::' ':::::.
     *       .::'        :::::.:::::::::'      ':::::.
     *      .::'         ::::::::::::::'         ``::::.
     *  ...:::           ::::::::::::'              ``::.
     * ```` ':.          ':::::::::'                  ::::..
     *  		           '.:::::'          			':'````..
     */
    
    package main
    
    import "errors"
    
    type Animal struct {
    	AnimalType string
    }
    
    func (a *Animal) SetAnimalType(AnimalType string) {
    	a.AnimalType = AnimalType
    }
    
    func (a *Animal) GetAnimalType() string {
    	return a.AnimalType
    }
    
    type Dog struct {
    	a Animal
    }
    
    func (d *Dog) Dog() string {
    	d.a.SetAnimalType("dog")
    	return d.a.GetAnimalType()
    }
    
    type Cat struct {
    	a Animal
    }
    
    func (c *Cat) Cat() string {
    	c.a.SetAnimalType("cat")
    	return c.a.GetAnimalType()
    }
    
    type AnimalEnterQueue struct {
    	Animal Animal
    	Count  int
    }
    
    func (a *AnimalEnterQueue) Set(animal Animal, count int) {
    	a.Animal = animal
    	a.Count = count
    }
    
    func (a *AnimalEnterQueue) GetAnimal() Animal {
    	return a.Animal
    }
    
    func (a *AnimalEnterQueue) GetCount() int {
    	return a.Count
    }
    
    func (a *AnimalEnterQueue) GetAnimalType() string {
    	return a.Animal.AnimalType
    }
    
    type DogCatQueue struct {
    	DogQueue []AnimalEnterQueue
    	CatQueue []AnimalEnterQueue
    	Count    int
    }
    
    func (a *DogCatQueue) DogCatQueueAdd(animal Animal) (bool, error) {
    	if animal.GetAnimalType() == "dog" {
    		a.Count++
    		a.DogQueue = append(a.DogQueue, AnimalEnterQueue{animal, a.Count})
    		return true, nil
    	}
    
    	if animal.GetAnimalType() == "cat" {
    		a.Count++
    		a.CatQueue = append(a.CatQueue, AnimalEnterQueue{animal, a.Count})
    		return true, nil
    	}
    
    	return false, errors.New("AnimalType不存在")
    }
    
    func (a *DogCatQueue) IsEmpty() (bool, error) {
    	if len(a.DogQueue) == 0 && len(a.CatQueue) == 0 {
    		return true, errors.New("狗猫队列为空")
    	}
    
    	return false, errors.New("狗猫队列不为空")
    }
    
    func (a *DogCatQueue) IsDogQueueEmpty() (bool, error) {
    	if len(a.DogQueue) == 0 {
    		return true, errors.New("狗队列为空")
    	}
    
    	return false, errors.New("狗队列不为空")
    }
    
    func (a *DogCatQueue) IsCatQueueEmpty() (bool, error) {
    	if len(a.CatQueue) == 0 {
    		return true, errors.New("猫队列为空")
    	}
    
    	return false, errors.New("猫队列不为空")
    }
    
    func (a *DogCatQueue) DogCatQueuePollAll() (Animal, error) {
    	dogLength := len(a.DogQueue)
    	catLength := len(a.CatQueue)
    	if dogLength > 0 && catLength > 0 {
    		dogQueue := a.DogQueue[dogLength-1]
    		catQueue := a.CatQueue[catLength-1]
    		if dogQueue.GetCount() > catQueue.GetCount() {
    			a.DogQueue = a.DogQueue[0 : dogLength-1]
    			return dogQueue.GetAnimal(), nil
    		} else {
    			a.CatQueue = a.CatQueue[0 : catLength-1]
    			return catQueue.GetAnimal(), nil
    		}
    	}
    
    	if dogLength > 0 {
    		dogQueue := a.DogQueue[dogLength-1]
    		a.DogQueue = a.DogQueue[0 : dogLength-1]
    		return dogQueue.GetAnimal(), nil
    	}
    
    	if catLength > 0 {
    		catQueue := a.CatQueue[catLength-1]
    		a.CatQueue = a.CatQueue[0 : catLength-1]
    		return catQueue.GetAnimal(), nil
    	}
    
    	return Animal{}, errors.New("DogCatQueue不存在")
    }
    
    func (a *DogCatQueue) DogPollAll() (Animal, error) {
    	dogLength := len(a.DogQueue)
    	if dogLength == 0 {
    		return Animal{}, errors.New("DogQueue不存在")
    	}
    
    	dogQueue := a.DogQueue[dogLength-1]
    	if dogLength == 1 {
    		a.DogQueue = []AnimalEnterQueue{}
    	} else {
    		a.DogQueue = a.DogQueue[0 : dogLength-1]
    	}
    
    	return dogQueue.GetAnimal(), nil
    }
    
    func (a *DogCatQueue) CatPollAll() (Animal, error) {
    	catLength := len(a.CatQueue)
    	if catLength == 0 {
    		return Animal{}, errors.New("CatQueue不存在")
    	}
    
    	catQueue := a.CatQueue[catLength-1]
    	if catLength == 1 {
    		a.CatQueue = []AnimalEnterQueue{}
    	} else {
    		a.CatQueue = a.CatQueue[0 : catLength-1]
    	}
    
    	return catQueue.GetAnimal(), nil
    }