面试题 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;
}
};