两种方法实现,用两个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;
}
};
原文链接: https://www.cnblogs.com/ranjiewen/p/5474904.html
欢迎关注
微信关注下方公众号,第一时间获取干货硬货;公众号内回复【pdf】免费获取数百本计算机经典书籍
原创文章受到原创版权保护。转载请注明出处:https://www.ccppcoding.com/archives/233174
非原创文章文中已经注明原地址,如有侵权,联系删除
关注公众号【高性能架构探索】,第一时间获取最新文章
转载文章受原作者版权保护。转载请注明原作者出处!