洛谷P4799—— [CEOI2015 Day2]世界冰球锦标赛(折半搜索)

简介: 洛谷P4799—— [CEOI2015 Day2]世界冰球锦标赛(折半搜索)

原题链接

题意:

给n个物品和对应的价格,问用不超m元的钱可以买到的物品的方案数。

思路:

折半搜索的入门题(雾.jpg)

n的范围是n<=40。普通的搜索一定会TLE,考虑将物品分成两部分来搜索,这样时间复杂度就变成了2^(𝑛/2+1);

基本思路就是对[1,mid] 和[mid+1],n两个区间都进行dfs,分别记录答案,再将两个答案区间合并。

就本题而言,a数组表示对[1,mid]进行搜索,得到的方案;b数组表示对[mid+1,n]进行搜索,得到的方案。

区间合并时,对于a,b数组里的任意两个数,只要和<=m都是合法的方案数,朴素解法是两层for枚举,只能过40%的数据。优化是先将a数组从小到大排序,再遍历b数组,在a数组里二分查找b[i]+a[pos]<=m的最大的pos值,这样pos之前的数一定也满足该式。

代码:

注意数组大小。

const int maxn=1e7+100;
ll n,m;
ll w[maxn],a[maxn],b[maxn];
ll cnta=0,cntb=0;
void dfs(ll l,ll r,ll sum,ll a[],ll &cnt){
    if(sum>m) return ;///剪枝
    if(l>r){
        a[++cnt]=sum;return ;///搜索完成
    }
    dfs(l+1,r,sum+w[l],a,cnt);///选当前数
    dfs(l+1,r,sum,a,cnt);///不选当前数
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++) w[i]=read();
    ll mid=n/2;///折半搜索
    dfs(1,mid,0,a,cnta);
    dfs(mid+1,n,0,b,cntb);
    sort(a+1,a+1+cnta);///对a数组进行排序
    ll res=0;
    for(int i=1;i<=cntb;i++){
        ///遍历b数组 看b[i]和前面的几个数的和不超过m
        res+=upper_bound(a+1,a+1+cnta,m-b[i])-(a+1);
    }
    out(res);
    return 0;
}

参考

目录
相关文章
|
算法 索引
【算法挨揍日记】day09——35. 搜索插入位置、69. x 的平方根
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
89 0
|
7月前
|
存储
【洛谷 P1255】数楼梯 题解(递归+记忆化搜索+高精度)
这是一个使用动态规划解决的“数楼梯”问题。给定楼梯数`N`,求不同上楼方式的数量。程序通过递归函数`f()`计算,其中`f(x) = f(x - 1) + f(x - 2)`,初始条件`f(1) = 1`,`f(2) = 2`。考虑到数据规模可能很大,使用了高精度加法运算。样例输入`4`,输出`5`。代码中定义了一个存储中间结果的向量`mem`,并提供了一个用于显示高精度数的`printv()`函数。
128 0
|
7月前
|
存储
【洛谷 P2437】蜜蜂路线 题解(递归+记忆化搜索+高精度)
蜜蜂路线问题:蜜蜂从蜂房$m$到$n$($m&lt;n$)按数字递增爬行。给定$m$和$n$,求路线数。示例:$m=1$,$n=14$,输出$377$。100%数据$1\leq m,n\leq1000$。使用斐波那契序列优化,高精度处理大数。代码实现斐波那契存储并动态规划求解。
114 0
|
7月前
【Leecode刷题】二分查找:x的平方根、搜索插入位置
【Leecode刷题】二分查找:x的平方根、搜索插入位置
|
算法 Android开发
LeetCode 周赛上分之旅 #38 结合排序不等式的动态规划
学习数据结构与算法的关键在于掌握问题背后的算法思维框架,你的思考越抽象,它能覆盖的问题域就越广,理解难度也更复杂。在这个专栏里,小彭与你分享每场 LeetCode 周赛的解题报告,一起体会上分之旅。
90 0
[leetcode 324] 摆动排序 II 思维+排序
[leetcode 324] 摆动排序 II 思维+排序
88 0
|
机器学习/深度学习 算法 Java
刷爆 LeetCode 周赛 337,位掩码/回溯/同余/分桶/动态规划·打家劫舍/贪心
上周末是 LeetCode 第 337 场周赛,你参加了吗?这场周赛第三题有点放水,如果按照题目的数据量来说最多算 Easy 题,但如果按照动态规划来做可以算 Hard 题。
125 0
|
机器学习/深度学习 存储 算法
搜索与图论 - floyd 算法
搜索与图论 - floyd 算法
|
算法 JavaScript 前端开发
日拱算法:双指针解“判断子序列”,除夕快乐~
算法继续,本篇带来的是非常典型的一道题:“判断子序列”,采用的是双指针的解法~
|
人工智能
LDUOJ——I. 买汽水(折半搜索+双指针)
LDUOJ——I. 买汽水(折半搜索+双指针)
116 0