买瓜 蓝桥杯 dfs 剪枝

三种情况：1.不要当前瓜，2.要当前整个瓜，3.要当前瓜的一半，减枝：如果当前总和大了或者太小了，都返回，怎么判断太小了？用一个后缀数组，如果sum + b[u] < m，那么就太小了，加上后面的数也不够

#include<iostream>
#include<algorithm>
using namespace std;
double a[35], b[35];
int n, m, ans = 1e9, f;

bool cmp(const double &p, const double &q){
    return p > q;
}

/*
 * u: 表示当前正在处理的瓜的编号。
 * cnt: 表示到目前为止已经切了几刀。
 * sum: 表示到目前为止所有被切的瓜的总重量。
 */

//第几个瓜，切了几刀，总和多少
void dfs(int u, int cnt, double sum){

    /*
     * 如果当前的总重量加上剩余的瓜的重量仍然无法达到目标重量m
     * 那么就没有必要继续搜索，可以直接返回。
     */
    
    //如果当前总和大了，或者小了，就减枝
    if(sum > m || sum + b[u] < m) return;
    if(sum == m){
        f = 1;
        ans = min(ans, cnt);
        return;
    }
    //这个判断要放在后面，防止最后一个数没算上就返回了
    if(u > n) return;
    dfs(u + 1, cnt, sum + a[u]);
    dfs(u + 1, cnt + 1, sum + a[u] / 2);
    dfs(u + 1, cnt, sum);
}

int main()
{
    cin>>n>>m;//n个瓜，m的重量
    for(int i = 1; i <= n; i++)
    {
        cin>>a[i];
    }

    /*
     * 为什么需要降序排列：
     * 相当于优先拿大石头把瓶子塞得差不多，然后填体积小的沙子进去
     */

    //降序排列，先加重量大的
    sort(a + 1, a + n + 1, cmp);
    //存储后缀和，用来减枝
    for(int i = n; i > 0; i--)
    {
        b[i] = b[i + 1] + a[i];
    }
    dfs(1, 0, 0);
    if(!f) printf("-1");
    else printf("%d", ans);
    return 0;
}