蓝桥杯 试题 算法训练 车的放置

资源限制

内存限制：256.0MB   C/C++时间限制：1.0s   Java时间限制：3.0s   Python时间限制：5.0s

问题描述

　　在一个n*n的棋盘中，每个格子中至多放置一个车，且要保证任何两个车都不能相互攻击，有多少中放法(车与车之间是没有差别的)

输入格式

　　包含一个正整数n

输出格式

　　一个整数，表示放置车的方法数

样例输入

2

样例输出

7

数据规模和约定

　　n<=8
　　【样例解释】一个车都不放为1种，放置一个车有4种，放置2个车有2种。

集体代码如下：

#include <iostream>
using namespace std;
int a[9][9]={0},n;//数组a代表了棋盘，若位置元素上为0，则说明未落子，若为1，则说明落子，n代表了棋盘宽度
int sum=0;//sum用于记录否可行方法总和
bool judge(int z)//定义judge函数判断该位置是否能够落子
{
	int x=0,y=0;
	while(x<(z/n))//试探z位置所在行是否存在落子，如果存在，返回false（此时z位置所在列为z%n）
	{//z位置下方位置还未试探到不需要判断
		if(a[x][z%n]==1)
			return false;
		x++;
	}
	while(y<(z%n))//试探z位置所在列是否存在落子，如果存在，返回false(此时z位置所在行为z/n)
	{//z位置右方位置还未试探到不需要判断
		if(a[z/n][y]==1)
			return false;
		y++;
	}
	return true;
}
void search(int z=0)//这里的z代表了第z个位置(二维数组实际上是一维数组分成了多段，叠起来),z/n代表该位置所在行数，z%n代表该位置所在列数
{
	if(z>n*n-1)//如果试探完棋盘所有位置，那么该方法可行，那么让sum+1
	{
		sum++;
		return;
	}
	search(z+1);//这里我们直接试探下个位置，当前z位置未落子
	if(judge(z)==true)//我们使用judge函数来判断当前是否可以落子
	{
		a[z/n][z%n]=1;//假设该位置落子
		search(z+1);//试探下个位置是否可以落子
		a[z/n][z%n]=0;//试探完后进行还原
	}
}
int main()
{
	cin>>n;//输入棋盘宽度
	search();
	cout<<sum;
	return 0;
}

欢迎大家留言评论，谢谢大家！