面试题 03.06. 动物收容所

面试题 03.06. 动物收容所

思路：队列，记录猫和狗，all记录当前最老的是猫还是狗

class AnimalShelf {
public:
    queue<vector<int> > cat, dog;
    vector<int> all;
    AnimalShelf() {
        while (!cat.empty()) cat.pop();
        while (!dog.empty()) dog.pop();
        all.clear();
    }
    
    void enqueue(vector<int> animal) {
        if(animal[1]==0) {cat.push(animal);all.push_back(0);}
        else {dog.push(animal);all.push_back(1);}
    }
    
    vector<int> dequeueAny() {
        if(all.size()==0) return {-1,-1};
        if(all[0]==0) return dequeueCat();
        return dequeueDog();
    }
    
    vector<int> dequeueDog() {
        if(dog.size()==0) return {-1,-1};
        for(int i=0;i<all.size();i++){
            if(all[i]==1){
                all.erase(all.begin()+i);
                break;
            }
        }
        vector<int> res = dog.front();
        dog.pop();
        return res;
    }
    
    vector<int> dequeueCat() {
        if(cat.size()==0) return {-1,-1};
        for(int i=0;i<all.size();i++){
            if(all[i]==0){
                all.erase(all.begin()+i);
                break;
            }
        }
        vector<int> res = cat.front();
        cat.pop();
        return res;
    }
};