P1036 [NOIP2002 普及组] 选数

最新推荐文章于 2024-11-04 19:24:37 发布

天火小周周

最新推荐文章于 2024-11-04 19:24:37 发布

阅读量144

点赞数 3

文章标签：深度优先算法

本文链接：https://blog.csdn.net/qq_62809112/article/details/123150911

版权

P1036 [NOIP2002 普及组] 选数

题目链接
https://www.luogu.com.cn/problem/P1036

在这里插入图片描述

题目分析

类似全排列，但是考虑到有重复的问题，需要去重
这里我运用了sc 表示（开始查找的位置），每次将i+1的位置标记为下次查找的开始

代码

#include<iostream>
using namespace std;
int n, k;
int a[101];
bool vis[101];
int book[1001];
int ans = 0;
bool zhishu(int k)
{
	if (k < 2)return 0;
	for (int i = 2; i * i <= k; i++)
		if (k % i == 0)return 0;
	return 1;
}
void dfs(int step,int sc)
{
	if (step == k)
	{
		int sum = 0;
		for (int i = 0; i < k; i++)
			sum += book[i];
		if (zhishu(sum))
			ans++;
		return;
	}
	for(int i=sc;i<n;i++)
		if (!vis[i])
		{
			vis[i] = 1;
			book[step] = a[i];
			dfs(step + 1,i+1);
			vis[i] = 0;
		}
}
int main()
{
	cin >> n >> k;
	for (int i = 0; i < n; i++)
		cin >> a[i];
	dfs(0,0);
	cout << ans;
	return 0;
}