04-狐狸吃兔子（C++）

Problem

Description

狐狸吃兔子：某地有N个洞（0,1,2,3…,N-1），狐狸住第0号，狐狸每次从左向右搜索一次，搜索方法为：
正向搜索：先从1号洞，隔一个，搜索3号洞，然后隔两个搜索6号洞，然后隔3个搜索10号洞，以此类推。
逆向搜索：先从N-1号洞，隔一个，搜索N-3号洞，然后隔两个搜索N-6号洞，然后隔3个搜索N-10号洞，以此类推。
狐狸搜索过的洞兔子都不能住，请输出兔子可以住的洞。

Input

一个正整数N,表示总的洞穴数量。

Output

第一行：兔子可以住的洞穴编号，以空格分隔。
第二行：兔子可以住的洞穴数量

Code

代码一

使用malloc()进行动态内存分配，这种方法耗费空间。

#include<iostream>
using namespace std;

int main()
{
	int N;//总的洞穴数量
	int s = 2;
	int count = 0;//记录兔子可以住的洞的数量

	cin >> N;
	bool *flag=(bool*)malloc(N);//能存放N个元素的数组
	for (int i = 1; i < N; i++)
		flag[i] = true;//初始化为true,true表示狐狸没有搜索过

	for (int i = 1, j = N - 1; i < N; i += s, j -= s,s++)//标记狐狸搜索过的洞
	{
		flag[i] = false;
		flag[j] = false;
	}
	for (int i = 2; i < N - 1; i++)//检查哪些洞兔子可以住
	{
		if (flag[i] == true)
		{
			count++;
			cout << i << ' ';//输出兔子可以住的洞的编号
		}	
	}
	cout << endl;
	cout << count << endl;
	return 0;
}

代码二

利用循环，检测每个洞穴 是否被正向搜索过 或者 是否被逆向搜索过。

#include<iostream>
using namespace std;

/*
* 正向搜索到的洞穴编号（是一个累加的过程：1+2+3+4+...）：
* 1
* 1+2 = 3
* 1+2+3 = 6
* 1+2+3+4 = 10
* ...
* 逆向搜索到的洞穴编号：
* N-1
* N-1-2 = N-3
* N-1-2-3 = N-6
* N-1-2-3-4 = N-10
* ...
*/

int main()
{
	int N;//输入的洞穴数量
	int count = 0;//记录兔子可以住的洞的数量

	cin >> N;

	int d = 2;//步长
	for (int i = 1; i < N; i++)
	{
		bool flag = true;
		for (int j = 1; j < N;)//正向搜索
		{
			if (j == i)//洞被狐狸搜索过，则flag置为false
			{
				flag = false;
				break;
			}
			if (j > i)//洞没有被搜索过，提前结束，减少循环，减少运行时间
				break;
			j = j + d;//搜索到的下一个洞穴的编号
			d++;
		}
		if (flag == true) {//正向搜索没有搜索过的洞穴，才检查逆向搜索
			d = 2;
			for (int j = N - 1; j < N;)//逆向搜索
			{
				if (j == i)
				{
					flag = false;
					break;
				}
				if (j < i)//j<i，洞没有被搜索过，提前结束循环
					break;
				j = j - d;//搜索到的下一个洞穴的编号
				d++;
			}
		}
		d = 2;//重置步长
		if (flag == true) {
			count++;
			cout << i << ' ';
		}
	}
	cout << endl;
	cout << count << endl;
	return 0;
}

若有错误请留言。