猫狗收容所

  • 有家动物收容所只收留猫和狗,但有特殊的收养规则,收养人有两种收养方式,第一种为直接收养全部动物中最先进入收容所的,第二种为选择收养的动物类型(猫或狗),并收养该种动物中最先进入收容所的。 给定一个操做序列int[][2] ope(C++中为vector<vector<int>>)表明全部事件。若第一个元素为1,则表明有动物进入收容所,第二个元素为动物的编号,正数表明狗,负数表明猫;若第一个元素为2,则表明有人收养动物,第二个元素若为0,则采起第一种收养方式,若为1,则指定收养狗,若为-1则指定收养猫。请以顺序返回收养的序列。若出现不合法的操做,即没有能够符合领养要求的动物,则将此次领养操做忽略。
  • 测试样例:

[[1,1],[1,-1],[2,0],[2,-1]]c++

  • 返回:

[1,-1]测试

  • 解答: 根据先进先出的原则,天然想到了用队列实现,若是只维护一个队列,那么第一种方法实现起来比较简单,只须要取出队头的动物则能够,可是第二种方法就比较复杂,须要访问整个队列来找出第一个被访问的猫或者狗。
  • 所以咱们能够选择维护两个队列来实现,一个队列存放放入的狗,一个队列存放放入的猫,对于第二种方法实现起来至关容易,咱们只须要根据要选择的猫或者狗从相应的队列中取出即可以,可是第一种方法须要判断那个两个队列头部的是猫先进入收容所,仍是狗先进入,这个时候须要一个标志,因此咱们每次把动物放入队列的时候,同时将一个递增的序号放入队列,这个序号就是一个时间序列,根据这个序号即可以轻松实现第一种方法。
class CatDogAsylum {
public:
    vector<int> asylum(vector<vector<int> > ope) {
        // write code here
        queue<int> cat;
        queue<int> dog;
        vector<int> vec;
        int index=0;
        int size1=ope.size();
        for(int i=0;i<size1;i++)
        {
            int kind=ope[i][0];
            if(kind==1)
            {
                if(ope[i][1]>=0)
                {
                    dog.push(index++);
                    dog.push(ope[i][1]);
                }
                else
                {
                    cat.push(index++);
                    cat.push(ope[i][1]);
                }
            }
            else
            {
                if(ope[i][1]==0)
                {
                    int min=0;
                    if(cat.empty()&&!dog.empty())
                        min=1;
                    if(!cat.empty()&&dog.empty())
                        min=-1;
                    if(!cat.empty()&&!dog.empty())
                        min=dog.front()>cat.front()?-1:1;
                    if(min==-1)
                    {
                        cat.pop();
                        vec.push_back(cat.front());
                        cat.pop();
                    }
                    if(min==1)
                    {
                        dog.pop();
                        vec.push_back(dog.front());
                        dog.pop();
                    }
                }
                else
                {
                    if(ope[i][1]==1&&!dog.empty())
                    {
                        dog.pop();
                        vec.push_back(dog.front());
                        dog.pop();
                    }
                    if(ope[i][1]==-1&&!cat.empty())
                    {
                        cat.pop();
                        vec.push_back(cat.front());
                        cat.pop();
                    }
                }
 
            }
        }
        return vec;
    }
};

吐槽一下,题干太长了code

相关文章
相关标签/搜索