和为 k 的组合(51Nod-1268)
生活随笔
收集整理的這篇文章主要介紹了
和为 k 的组合(51Nod-1268)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
題目
給出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 標記會超時,因此可以對搜索順序進行優化,每次 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)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 信息学奥赛一本通(1031:反向输出一个
- 下一篇: Color the ball(HDU-1