DFS 部分和问题

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

限制条件

1<=n<=20

-10^n<=ai<=10^8

-10^8<=k<=10^8

输入:

n=4

a={1,2,4,7}

k=13

输出:

Yes (13 = 2 + 4 + 7)

int a[MAX_N];
int n, k;

// 已经从i项得到了和sum, 然后对于i项之后的进行分支
bool dfs(int i, int sum)
{
    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;

    return false;
}

void solve()
{
    if(dfs(0, 0)) printf("Yes\n");
    else printf("No\n"); 
}
发布了58 篇原创文章 · 获赞 6 · 访问量 7061

猜你喜欢

转载自blog.csdn.net/weixin_43569916/article/details/103939992