一道DFS的小题: 部分和问题

部分和问题  (挑战程序设计竞赛)


给定整数 a 1 、a 2 、…、a n ,判断是否可以从中选出若干数,使它们的和恰好为 k。 


输入 
n=4 
a={1,2,4,7} 
k=13 
输出 
Yes (13 = 2 + 4 + 7) 


输入 
n=4 
a={1,2,4,7} 
k=15 
输出 
No 


// 输入 
int a[MAX_N]; 
int n, k; 
 
// 已经从前i项得到了和sum,然后对于i项之后的进行分支 
bool dfs(int i, int sum) { 
  // 如果前n项都计算过了,则返回sum是否与k相等 
  if (i == n) return sum == k;  
   
  // 不加上a[i]的情况 
  if (dfs(i + 1, sum)) return true; 
   
  // 加上a[i]的情况 
  if (dfs(i + 1, sum + a[i])) return true; 
   
  // 无论是否加上a[i]都不能凑成k就返回false 
  return false; 
} 
 
void solve() { 
  if (dfs(0, 0)) printf("Yes\n"); 
  else printf("No\n"); 
} 


心血来潮想画出DFS调用树


一道DFS的小题: 部分和问题_第1张图片

你可能感兴趣的:(一道DFS的小题: 部分和问题)