P1036 [NOIP2002 普及组] 选数 题解

[NOIP2002 普及组] 选数

题目描述

已知 $n$ 个整数 $x_1,x_2,\cdots ,x_n$，以及 $1$ 个整数 $k$（$k<n$）。从 $n$ 个整数中任选 $k$ 个整数相加，可分别得到一系列的和。例如当 $n=4$，$k=3$，$4$ 个整数分别为 $3,7,12,19$ 时，可得全部的组合与它们的和为：

$3+7+12=22$

$3+7+19=29$

$7+12+19=38$

$3+12+19=34$

现在，要求你计算出和为素数共有多少种。

例如上例，只有一种的和为素数：$3+7+19=29$。

输入格式

第一行两个空格隔开的整数 $n,k$（$1\le n\le 20$，$k<n$）。

第二行 $n$ 个整数，分别为 $x_1,x_2,\cdots ,x_n$（$1\le x_i\le 5\times 10^6$）。

输出格式

输出一个整数，表示种类数。

样例 #1

样例输入 #1

4 3
3 7 12 19

样例输出 #1

1

提示

【题目来源】

NOIP 2002 普及组第二题

题解

这个题看起来很简单，实际也很简单，但是要注意这是第二题的位置上，难度在数据规模上
如果搜索，时间复杂度按20的阶乘算肯定超了，但是这个题本身是组合题，如果最坏情况20选10是组合，那么计算次数绝对可以接受（按按计算器就行了），所以全部的难点在于避重
我没有用搜索的形式，而是用了k个指针（不是C++语法的指针），模拟纸上用手枚举
几点性质：

1. 静态的性质：对于这k个指针，每一个指针都有自己的结束位置
2. 动态的性质：指针每次移动的时候，从右向左更新，即先看后面的指针能不能移动
3. 动态的性质：如果不是末尾指针的更新，更新完毕后，需要让该指针的后续指针归位，归位到紧挨在该指针后面。且指针的更新无需担心越界，每一个指针都有属于自己的结束位置，某指针只要更新时不超过自己的结束位置，某指针的后面的指针归位时无需担心会因此越界。

代码如下：

#include<iostream>
using namespace std;
int n,k,num[21],pt[21],ans;
bool check_end() //检查是否枚举结束
{
	for(int i=1;i<=k;i++)
	if(pt[i]!=n+i-k) return false;
	return true;
}
bool check_prime(int x) //检查结果是否为质数
{
	for(int i=2;i*i<=x;i++)
	if(x%i==0) return false;
	return true;
}
int sum() //求和
{
	int tmp=0;
	for(int i=1;i<=k;i++)
	{
		tmp+=num[pt[i]];
	}
	return tmp;
}
void pt_move() //移动指针
{
	for(int i=k;i>=1;i--)
	{
		if(pt[i]<n+i-k) { //每一个指针都有对应的结束位置
			pt[i]++;
			for(int j=1;j<=k-i;j++)
			pt[i+j]=pt[i]+j; //更新后面的指针归位
			break;
		}
	}
}
void print() //调试用
{
	for(int i=1;i<=k;i++)
	cout<<pt[i]<<" ";
	cout<<endl;
	ans+=1;
}
int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++)
	cin>>num[i];
	
	//指针初始位置
	for(int i=1;i<=k;i++)
	pt[i]=i;
	
	while(1)
	{
		if(check_prime(sum())) ans++; //计数
		if(check_end()) break; //判断是否枚举结束
		pt_move(); //移动指针
	}
	cout<<ans<<endl;
	return 0;
}