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