[保研/考研机试] 猫狗收容所 C++实现

题目描述:

输入:

第一个是n,它代表操作序列的次数。接下来是n行,每行有两个值m和t,分别代表题目中操作的两个元素。

输出:

按顺序输出收养动物的序列,编号之间以空格间隔。

源代码:

#include
#include
#include
using namespace std;

//例题5.3 猫狗收容所
int main()
{
	int n, m, t, first=0;
	cin >> n;
	//定义两个队列,分别存储mao和狗进入收养所的顺序和编号,第一个进入的除外
	queue cat;
	queue dog;
	//按顺序保存出收养所的动物编号
	vector nums;
	for (int i = 0; i < n; i++) { //循环
		cin >> m >> t;
		//保存第一个进入收容所的动物的编号。因为这里只需要输出出收养所的动物编号的顺序,如果把第一个进入收养所的也如队列的话,
		//后续输出第一个进入收养所的动物编号比较麻烦,直接通单独的一个变量保存更方便
		if (first == 0 && m==1) { 
			first = t;
			continue;
		}
		if (m == 1) { // 有动物进入收养所
			if (t > 0) { //进入收养所的是狗
				dog.push(t);
			}
			if (t < 0) { //进入收养所的是猫
				cat.push(t);
			}
		}
		if (m == 2) { // 有人收养动物
			if (t == 0 && first != 0) { //收养最早进入的
				nums.push_back(first);
				//cout << first << " ";
			}
			if (t == 1 && !dog.empty()) { //收养狗
				//cout << dog.front() << " ";
				nums.push_back(dog.front());
				dog.pop();
			}
			if (t == -1 && !cat.empty()) { //收养猫
				//cout << cat.front() << " ";
				nums.push_back(cat.front());
				cat.pop();
			}
		}
	}
	//按顺序输出出收养所的动物编号
	for (int i = 0; i < nums.size()-1; i++) {
		cout << nums[i] << " ";
	}
	cout << nums[nums.size() - 1] << endl;


	return 0;
}

运行结果:

[保研/考研机试] 猫狗收容所 C++实现_第1张图片

 

你可能感兴趣的:(机试刷题记录,算法,c++,数据结构,研究生复试)