2019深信服笔试 —— 猎人抓兔子

2019深信服笔试 —— 猎人抓兔子

题目描述:

       假设兔子有n个排成一排的洞，编号为1和n。兔子每天晚上跳到相邻的一个洞里住，而猎人每天白天检查其中的一个洞。猎人告诉你每天检查的洞的编号，希望你分析一下是否一定能抓到兔子。

输入描述：

       首先输入n和k（n,k<1000），分别表示洞的个数和猎人要检查的天数；接下来输入k个整数，表示猎人每天检查的洞的编号。

输出描述：

       对于每个样例，如果一定能抓到兔子，输出Yes，否则输出No。

示例：

样例1：

输入：

3  2

2  2

输出：

Yes

思路分析：

       首先将所有洞口的值置为1，表示兔子可能在此洞中；假设猎人每次检查洞口都抓不到兔子(兔子不在此洞中)，将该洞口的值置为0；再将每天晚上兔子可能跳到的洞口值置为1，不可能跳到的洞口值置为0，当所有的洞口全为0的时候兔子无洞可逃，则肯定会被猎人抓住。

代码如下：

#include<iostream>
#include<vector>
using namespace std;

void print(vector<int> ivec)
{
	for (size_t i = 0; i < ivec.size(); i++)
	{
		cout << ivec[i] << " ";
	}
	cout << endl;
}
bool canCatchRibbit(vector<int> holeNums, vector<int> checked)
{
	int k = holeNums.size();
	int n = checked.size();
	bool flag = true;
	int cnt = n;  //  cnt为兔子能待的洞的个数，初试化为所有洞的个数
	for (int i = 0; i < k; i++)
	{
		int holeNum = holeNums[i] - 1; //  holeNum为洞的编号在vector中的序号，
		                               //  如：编号为1的洞在vector里面存的是0
		if (checked[holeNum] == 1)
		{
			checked[holeNum] = 0;
			cnt--;
		}
		print(checked);
		cout << "----------------" << endl;
		if (cnt == 0)   //  如果兔子能待的洞的个数为0，则兔子肯定会被逮住
		{
			flag = false;
			break;
		}
		int temp = checked[n - 2];
		for (int j = 0; j < n; j++)
		{
			if (checked[j] == 1)
			{
				checked[j] = 0;
				cnt--;
				if (j>0 && checked[j - 1] == 0)
				{
					checked[j - 1] = 1;
					cnt++;
				}
				if (j < n - 1 && checked[j + 1] == 0)
				{
					checked[j + 1] = 1;
					cnt++;
					j++;  //  原来值为0的洞现在为1后是不能往相邻的洞跳，故继续检查下一个洞
				}
				if (j == n - 1 && temp == 1)   //  处理最后一个洞口值
				{
					checked[j] = 1;
					cnt++;
				}
			}
		}
		print(checked);
	}
	return flag;
}

int main()
{
	int n, k;  //  n为洞的个数，k为检查的天数
	cin >> n >> k;
	vector <int> holeNums(k, 0);  //  holeNums为每天检查的洞的编号
	for (auto &m : holeNums)
		cin >> m;
	vector<int> checked(n, 1);  //  checked为每天检查的洞中是否有兔子,可能有兔子则值为1，
	                            //  一定没兔子则值为0，全部初始化为1；
	cout << "----------------" << endl;
	print(checked);
	bool flag = canCatchRibbit(holeNums, checked);
	if (!flag)
		cout << "Yes" << endl;
	else
		cout << "No" << endl;
	return 0;
}

运行结果如下：