1. 程式人生 > >CODEVS 1008選數

CODEVS 1008選數

stream 範圍 滿足 都是 sam scrip 屏幕 body blog

題目描述 Description

已知 n 個整數 x1,x2,…,xn,以及一個整數 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)。

輸入描述 Input Description

 鍵盤輸入,格式為:
  n , k (1<=n<=20,k<n)
  x1,x2,…,xn (1<=xi<=5000000)

輸出描述 Output Description

屏幕輸出,格式為:
  一個整數(滿足條件的種數)。

樣例輸入 Sample Input

4 3
3 7 12 19

樣例輸出 Sample Output

1

數據範圍及提示 Data Size & Hint

(1<=n<=20,k<n)
(1<=xi<=5000000)

思路:這類似於求n個數中選k個數的組合問題。 將選出的k個數相加判斷是否是素數,如果是ans++。 這道題還有一個坑點是所給出的數全都是1,所以這裏需要加一個特判 代碼:
 1 #include<iostream>
 2
#include<cstdio> 3 #include<cmath> 4 using namespace std; 5 int n,k,ans; 6 int num[30],sum[30]; 7 bool vis[30]; 8 bool same() 9 { 10 int x=num[1]; 11 for(int i=2;i<=n;i++) 12 if(x!=num[i])return 0; 13 return 1; 14 } 15 bool prime(int x) 16 { 17 if(x==2)return
1; 18 for(int i=2;i<=sqrt(x);i++) 19 if(x%i==0)return 0; 20 return 1; 21 } 22 void dfs(int x) 23 { 24 if(x>k) 25 { 26 int cnt=0; 27 for(int i=1;i<=k;i++) 28 cnt+=sum[i]; 29 if(prime(cnt))ans++; 30 } 31 else 32 for(int i=1;i<=n;i++) 33 if(num[i]>sum[x-1] && !vis[i]) 34 { 35 sum[x]=num[i]; 36 vis[i]=1; 37 dfs(x+1); 38 vis[i]=0; 39 } 40 } 41 int main() 42 { 43 cin>>n>>k; 44 for(int i=1;i<=n;i++) 45 cin>>num[i]; 46 dfs(1); 47 if(same() && prime(num[num[1]*k])) 48 { 49 int a=1,b=1; 50 for(int i=n-k+1;i<=n;i++) 51 a*=i; 52 for(int i=1;i<=k;i++) 53 b*=i; 54 cout<<a/b; 55 return 0; 56 } 57 cout<<ans; 58 return 0; 59 }

CODEVS 1008選數