題目

給出N個正整數組成的數組A，求能否從中選出若干個，使他們的和為K。如果可以，輸出："Yes"，否則輸出"No"。

輸入

第1行：2個數N, K, N為數組的長度, K為需要判斷的和(2 <= N <= 20，1 <= K <= 10^9)
第2 - N + 1行：每行1個數，對應數組的元素A[i] (1 <= A[i] <= 10^6)

輸出

如果可以，輸出："Yes"，否則輸出"No"。

輸入樣例

5 13
2
4
6
8
10

輸出樣例

No

思路：dfs

普通的深搜加 vis 標記會超時，因此可以對搜索順序進行優(yōu)化，每次 dfs 在當前結點的下一個結點開始搜索即可

源程序

#include<iostream> #include<cstdio> #include<cstdlib> #include<string> #include<cstring> #include<cmath> #include<ctime> #include<algorithm> #include<utility> #include<stack> #include<queue> #include<vector> #include<set> #include<map> #include<bitset> #define EPS 1e-9 #define PI acos(-1.0) #define INF 0x3f3f3f3f #define LL long long const int MOD = 1E9+7; const int N = 1000000+5; const int dx[] = {-1,1,0,0,-1,-1,1,1}; const int dy[] = {0,0,-1,1,-1,1,-1,1}; using namespace std; int n,k; int a[N]; bool flag; void dfs(int sum,int step) {if(flag)return;if(sum==0) {flag=true;return;}if(step<=n){for(int i=step;i<=n;i++){sum-=a[i];dfs(sum,i+1);sum+=a[i];}} } int main() {scanf("%d%d",&n,&k);for(int i=1; i<=n; i++)scanf("%d",&a[i]);for(int i=1;i<=n;i++)dfs(k,i);if(flag)printf("Yes\n");elseprintf("No\n");return 0; }

?

總結

以上是生活随笔為你收集整理的和为 k 的组合（51Nod-1268）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。