首页 > 代码库 > 折半枚举 双向搜索
折半枚举 双向搜索
从四个数列选择的话共有n的4次方种情况。将它们对半分成ab和cd再考虑以快速解决。
从两个数列中选择只有n的2次方种情况,可以进行枚举。
从c和d中取数字的n的平方种方法全都枚举出来排好序。运用二分搜索从中搜索-a[i]-b[j]。复杂度O(n^2logn)。
#include<iostream> #include<math.h> #include<algorithm> #include<stdio.h> using namespace std; int n; int a[4005],b[4005],c[4005],d[4005]; int cd[16000005];//c和d中数字的组合方法 int main() { cin>>n; for(int i=0;i<n;i++) { cin>>a[i]>>b[i]>>c[i]>>d[i]; } long long res=0; for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ cd[i*n+j]=c[i]+d[j]; } } sort(cd,cd+n*n); for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ //取出c和d中和为 -a[i]-b[j]的部分 res+=upper_bound(cd,cd+n*n,-a[i]-b[j])-lower_bound(cd,cd+n*n,-a[i]-b[j]); } } cout<<res<<endl; return 0; }
折半枚举 双向搜索
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。