【程序员面试金典】猫狗收容所

题目描述

        有家动物收容所只收留猫和狗，但有特殊的收养规则，收养人有两种收养方式，第一种为直接收养所有动物中最早进入收容所的，第二种为选择收养的动物类型（猫或狗），并收养该种动物中最早进入收容所的。

       给定一个操作序列int[][2] ope(C++中为vector<vector<int>>)代表所有事件。若第一个元素为1，则代表有动物进入收容所，第二个元素为动物的编号，正数代表狗，负数代表猫；若第一个元素为2，则代表有人收养动物，第二个元素若为0，则采取第一种收养方式，若为1，则指定收养狗，若为-1则指定收养猫。请按顺序返回收养的序列。若出现不合法的操作，即没有可以符合领养要求的动物，则将这次领养操作忽略。

测试样例：

[[1,1],[1,-1],[2,0],[2,-1]]

返回：[1,-1]

---

class CatDogAsylum {
public:
    vector<int> asylum(vector<vector<int> > ope) {
        // write code here
        vector<int> A;   //定义动物序列
        vector<int> B;   //定义收养序列
        int i;     //定义循环变量
        int len = ope.size();
        for(i = 0; i < len; i ++)
        {
            if(ope[i][0] == 1) //有动物进来
                A.push_back(ope[i][1]);  //放入动物序列
            else
                if(A.empty())
                 continue;
                else
                 if(ope[i][1] == 0)  //第一种收养方式
                 {
                     B.push_back(A[0]);
                     A.erase(A.begin()); //从A中删除对应元素
                 }
                  else
                     if(ope[i][1] == 1)  //收养狗
                     {
                         for(int j = 0; j < A.size(); j ++)
                       if(A[j] > 0)
                       {
                        B.push_back(A[j]);
                        A.erase(A.begin() + j); //从A中删除对应元素
                        break;
                    }
                     }
                       else
                            for(int j = 0; j < A.size(); j ++)
                       if(A[j] < 0)
                       {
                        B.push_back(A[j]);
                        A.erase(A.begin() + j); //从A中删除对应元素
                        break;
                       }
        }
        return B;
    }
};