题目：数字王国之军训排队（蓝桥OJ 2942）

猫猫猫喵

已于 2024-03-17 15:12:01 修改

阅读量406

点赞数 6

分类专栏：算法竞赛文章标签：算法深度优先图论

于 2024-03-17 15:03:08 首次发布

本文链接：https://blog.csdn.net/2203_75582171/article/details/136782246

版权

算法竞赛专栏收录该内容

134 篇文章 0 订阅

订阅专栏

本文介绍了如何通过深度优先搜索（DFS）算法解决一个小规模的数据问题，即确定如何将1到n的学生分配到不超过n个队列中，使得队列内任意两个学生的分数互质。通过剪枝优化搜索过程。

摘要由CSDN通过智能技术生成

问题描述：

解题思路：

因为数据规模不大，因此我们可以从由1到10枚举全部可能答案，当符合时输出。使用dfs判断合法性，

题解：

#include<bits/stdc++.h>
using namespace std;
const int N = 15;
int a[N], n;
vector<int> v[N];  // v[i]：第i号队

bool dfs(int cnt, int dep)  // cnt：队伍数 dep：第几层/第几个元素 dfs：判断cnt时从dep开始是否合法
{
	if(dep == n + 1)
	{
		return true;
	}
  
	for(int i = 1; i <= cnt; i++)  // 枚举每个队伍，将当前dep的学生放到i号队伍中**
	{
	 	bool tag = true;
	 	for(const auto &j : v[i])	 // 剪枝：当i号队伍不能放时跳过到下一个队伍 
	 		if(a[dep] % j == 0 )
	 		{
	 			tag = false;
	 			break;
			}
		if(!tag)continue;


		v[i].push_back(a[dep]);  // 将dep层元素放入到i号队伍中
		if(dfs(cnt, dep + 1))return true;  // 如果合法返回true
			 
		v[i].pop_back();  // 恢复现场（如果dep学生放入i中会导致最终不合法则将其弹出）**
	}
	return false;  // 如果无法到最后一层则证明不合法，返回false
}

int main()
{
 	cin >> n;
 	for(int i = 1; i <= n; i++)cin >> a[i];
 	sort(a + 1, a + n + 1);  // 排序后则在出口用后面%前面 == 0，判断合法
	for(int i = 1; i <= n; i++)
	{
	 	if(dfs(i, 1))  
	 	{
	 		cout << i << '\n';
	 		break;
		}
	} 
	return 0;
}

知识点：dfs，剪枝