密码[双向搜索][STL[unordered_map]]

原创于 2018-11-01 21:47:38 发布 · 197 阅读

0 ·

CC 4.0 BY-SA版权

搜索专栏收录该内容

11 篇文章

订阅专栏

本文深入探讨了双向搜索算法的实现细节，通过将数组分为两部分并利用unordered_map存储中间结果，有效提高了查找效率。适用于解决特定类型的问题，如寻找数组中元素组合等于特定值k的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

双向搜索

将前一半所有的存下来, 在枚举后一半,枚举完后一半后找前一半的数组是否存在 k-(后一半)

存在,则ans加上(k-后一半)的个数

这里用unordered_map来存

unordered_map

#include<bits/stdc++.h>
#include<tr1/unordered_map>
#define N 50
#define ll long long 
using namespace std;
using namespace tr1;
int n,k; ll a[N],ans;
unordered_map<ll,int> S;
void dfs1(int u,int d,ll s){
	if(s>k) return;
	if(u>d){S[s]++;return;}
	dfs1(u+1,d,s); dfs1(u+1,d,s+a[u]);
}
void dfs2(int u,int d,ll s){
	if(s>k) return;
	if(u>d){
		if(S.find(k-s)!=S.end()) ans+=S[k-s]; return;
	}
	dfs2(u+1,d,s); dfs2(u+1,d,s+a[u]);
}
int main(){
	freopen("1.in","r",stdin);
	freopen("myself.out","w",stdout);
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	dfs1(1,n/2,0); dfs2(n/2+1,n,0);
	printf("%lld\n",ans); return 0;
}