貓狗收容所

  • 有家動物收容所只收留貓和狗,但有特殊的收養規則,收養人有兩種收養方式,第一種爲直接收養全部動物中最先進入收容所的,第二種爲選擇收養的動物類型(貓或狗),並收養該種動物中最先進入收容所的。 給定一個操做序列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

相關文章
相關標籤/搜索