洛谷-P1036 选数(dfs)
题目描述
已知 nnn 个整数 x1,x2,…,xnx_1,x_2,…,x_nx1,x2,…,xn,以及111个整数kkk(k<nk<nk<n)。从nnn个整数中任选kkk个整数相加,可分别得到一系列的和。例如当n=4,k=3n=4,k=3n=4,k=3,444个整数分别为3,7,12,193,7,12,193,7,12,19时,可得全部的组合与它们的和为:
3+7+12=223+7+12=223+7+12=22
3+7+19=293+7+19=293+7+19=29
7+12+19=387+12+19=387+12+19=38
3+12+19=343+12+19=343+12+19=34。
现在,要求你计算出和为素数共有多少种。
例如上例,只有一种的和为素数:3+7+19=293+7+19=293+7+19=29。
输入输出格式
输入格式:
1 #include<iostream> 2 #include<algorithm> 3 #include<cmath> 4 using namespace std; 5 typedef long long ll; 6 ll n, k, a[25],sum=0; 7 int s(ll x) { 8 int k = 0; 9 for (int i = 2;i < x;i++) { 10 if (x%i == 0)k++; 11 } 12 return k; 13 } 14 void dfs(ll x, ll y, ll z) { 15 if (x == n+1)return ; 16 if (z == k) { 17 if (s(y) == 0)sum++; 18 return ; 19 } 20 for (int i = x+1;i <= n;i++){ 21 dfs(i, y + a[i], z + 1); 22 23 } 24 //dfs(i, y + a[i], z+1); 25 } 26 int main() { 27 ios::sync_with_stdio(false); 28 cin >> n >> k; 29 for (int i = 1;i <=n;i++) 30 cin >> a[i]; 31 sort(a+1 , a + n+1); 32 dfs(0, 0, 0); 33 cout << sum<<endl; 34 return 0; 35 36 }
键盘输入,格式为:
n,kn,kn,k(1≤n≤20,k<n1 \le n \le 20,k<n1≤n≤20,k<n)
x1,x2,…,xn(1≤xi≤5000000)x_1,x_2,…,x_n (1 \le x_i \le 5000000)x1,x2,…,xn(1≤xi≤5000000)
输出格式:
屏幕输出,格式为: 111个整数(满足条件的种数)。