Fork me on GitHub

猫狗收养所

      两种方法实现,用两个vector实现,一个为领养的动物(输出),一个为收养的动物(输入);用两个queue实现(都为输入),一个为收养的猫,一个收养的狗,vetor(输出)。

/*****************************************************
* \file CatDogAsylum.cpp
* \date 2016/05/09 15:19

* \问题描述:
题目描述

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

* \问题分析:
思路:定义两个vector<int>对象A和B,分别用于存放收容所里的动物和被收养的动物
操作如下:
对ope数组从按行从i = 0 ~ ope.size() -1遍历
1、指令为1(ope[i][0] = 1)时,即有动物进来,则将动物序号压入A--执行A.push_back(ope[i][1]);
2、指令为2(ope[i][0] = 2)时,即有动物被收养,此时首先判断A是否为空,即是否有动物
1)如果A为空,则continue
2)如果A不为空
1.如果操作为ope[i][1] = 0,即收养最先进来的动物,则将A[0]压入B,执行B.push_back(A[0]),然后在A中删除对应元素,即执行A.erase(A.begin());
2.如果操作为ope[i][1] = 1,即收养最先进来的狗,此时遍历A找到第一个狗,然后将找到的元素压入B,再在A中删除对应元素;
3.如果操作为ope[i][1] = -1,即收养最先进来的猫,此时遍历A找到第一个猫,然后将找到的元素压入B,再在A中删除对应的元素。
遍历完成之后,返回B。


*****************************************************/
#include <iostream>
using namespace std;
#include <vector>
class CatDogAsylum {
public:
    vector<int> asylum(vector<vector<int> > ope) {
        // write code here
        int len = ope.size();
        vector<int> ans, in;
        if (len==0)
        {
            return ans;
        }
        for (int i = 0; i < len;i++)
        {
            if (ope[i][0]==1)  //有动物来收容所
            {
                in.push_back(ope[i][1]);
            }
            else if (ope[i][0]==2)  //    有人收养
            {
                if (in.empty())
                {
                    continue;
                }
                if (ope[i][1]==0)  //第一种收养方式,取第一个
                {
                    ans.push_back(in[0]);
                    in.erase(in.begin());
                }
                else if (ope[i][1]==1)  //指定养狗
                {
                    for (int i = 0; i < in.size();i++)
                    {
                        if (in[i]>0)
                        {
                            ans.push_back(in[i]);
                            in.erase(in.begin()+i);
                            break;
                        }
                    }
                }else if (ope[i][1]==-1) //指定养猫
                {
                    for (int i = 0; i < in.size();i++)
                    {
                        if (in[i]<0)
                        {
                            ans.push_back(in[i]);
                            in.erase(in.begin()+i);
                            break;
                        }
                    }
                }
            }
        }
        return ans;
    }
};


//////////////////////////////////////////////////////////////////////////
//题目分析:
//根据先进先出的原则,自然想到了用队列实现,如果只维护一个队列,那么第一种方法实现起来比较简单,只需要取出队头的动物则可以,
//但是第二种方法就比较复杂,需要访问整个队列来找出第一个被访问的猫或者狗。

//因此我们可以选择维护两个队列来实现,一个队列存放放入的狗,一个队列存放放入的猫,对于第二种方法实现起来相当容易,
//我们只需要根据要选择的猫或者狗从相应的队列中取出便可以,但是第一种方法需要判断那个两个队列头部的是猫先进入收容所,还是狗先进入,
//这个时候需要一个标志,所以我们每次把动物放入队列的时候,同时将一个递增的序号放入队列,这个序号就是一个时间序列,根据这个序号便可以轻松实现第一种方法。
//《程序员面试金典》--题目详解:
//http ://blog.csdn.net/zdplife/article/category/5799903
//////////////////////////////////////////////////////////////////////
#include <queue>
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())  //dog不为空
                        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;
    }
};

 

posted @ 2016-05-09 18:03  ranjiewen  阅读(777)  评论(0编辑  收藏  举报