2412 - 和为K ---深搜dfs剪枝

该代码示例展示了一种使用深度优先搜索(DFS)解决寻找数组中连续子数组和为K的优化方法。通过先对数组进行降序排序,可以减少无效的搜索。程序首先读取数组大小n和目标和k,然后对数组进行排序并反向,以利于DFS从较大数值开始。之后,利用滚动变量s记录前缀和,并进行两次DFS调用,分别考虑包含当前元素和不包含当前元素的情况。最后输出满足条件的子数组个数。
摘要由CSDN通过智能技术生成

**2412 - 和为K ---深搜dfs优化
**来源:东方博宜oj  oj.czos.cn
#include<bits/stdc++.h>
using namespace std;
const int N=30;
int a[N],s[N];
int n,k,ans=0;
void dfs(int idx,int sum)
{
    if(sum==k)
    {
        ans++;
        return;
    }
    if(sum>k) return;
    if(idx>n) return;
    if(sum+s[n]-s[idx-1]<k) return;
    dfs(idx+1,sum+a[idx]);
    dfs(idx+1,sum);
}
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)  cin>>a[i];
    //优化搜索顺序,从大到小数搜
    sort(a+1,a+1+n);
    reverse(a+1,a+1+n);
    for(int i=1;i<=n;i++) s[i]=s[i-1]+a[i];
    dfs(1,0);
    cout<<ans;
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值