hdu 5884(二分+优先队列)

//#if 0
#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <math.h>
#include <string>
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <time.h>
#include <set>
#include <list>
#include <iostream>
using namespace std;
#define ll long long
#define ull unsigned long long
#define cls(x) memset(x,0,sizeof(x))
#define clslow(x) memset(x,-1,sizeof(x))
#define INF  0x3f3f3f3f
const int maxn=1e5+50;

priority_queue<ll ,vector<ll>,greater<ll> >q;
ll n,t,x;
int num[maxn];

int check(ll k,ll lim)
{
    while(!q.empty())q.pop();
    ll ans=0,tmp;
//    cout<<"k=  "<<k<<endl;
    ll all=0;
    int flag=(n-1)%(k-1);
    if(flag>0){
        flag++;
        for(int i=1;i<=flag;i++) all+=num[i];
        q.push(all);
    }
    for(int i=flag+1;i<=n;i++)q.push(num[i]);
    while(q.size()>1){
        ll len=q.size();
//        cout<<" top = "<<q.top()<<endl;
        for(int i=1;i<=k;i++){
            tmp=q.top();
            q.pop();
            ans+=tmp;
        }
        all+=(ll)ans;
//        cout<<all<<endl;
        q.push(ans);
        ans=0;
    }
//    cout<<" ans = "<<all<<endl;
    if(all>lim)return 0;
    return 1;
}

int main()
{
//    freopen("in.txt","r",stdin);
    int t0;
    scanf("%d",&t0);
    while(t0--)
    {
        while(!q.empty())q.pop();
        scanf("%lld%lld",&n,&t);
        for(int i=1;i<=n;i++){
            scanf("%lld",&num[i]);
        }
        sort(num+1,num+1+n);
        ll l=2,r=n;
        while(l<r){
            ll mid=(l+r)/2;
            if(check(mid,t)==1){
                r=mid;
//            cout<<"mid == "<<mid<<endl;
            }
            else{
                l=mid+1;
//                cout<<mid<<endl;
            }
        }
        printf("%lld\n",l);
    }
    return 0;
}
//#endif

二分搜索2到n之间的答案,从2到n算出来的花费依次递减,但是因为其中有一种特例不会判断,即使自己想出来二分了,也不会做了。当n=7,k=5时,最少花费的合并应该是先合并最小的前三个值,然后入队,再合并剩余五个数,这样取得的花费最小;

鹅我就把前五个数合并再合并最后三个数。。。难受

大概思路就是用一个优先队列存值,每次取前mid个最小值合并完再入队,其中当出现n=7,k=5这种情况,判断就是flag=(n-1)%(k-1),当flag>0,则先合并前flag+1个值。然后当队列只有一个数的时候就可以和所给的 t 比较的了。

甚至还超时了一两次。。。。。。。注意一下把

不用long long 可以的

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
对于HDU4546问题,还可以使用优先队列(Priority Queue)来解决。以下是使用优先队列的解法思路: 1. 首先,将数组a进行排序,以便后续处理。 2. 创建一个优先队列(最小堆),用于存储组合之和的候选值。 3. 初始化优先队列,将初始情况(即前0个数的组合之和)加入队列。 4. 开始从1到n遍历数组a的元素,对于每个元素a[i],将当前队列中的所有候选值取出,分别加上a[i],然后再将加和的结果作为新的候选值加入队列。 5. 重复步骤4直到遍历完所有元素。 6. 当队列的大小超过k时,将队列中的最小值弹出。 7. 最后,队列中的所有候选值之和即为前k小的组合之和。 以下是使用优先队列解决HDU4546问题的代码示例: ```cpp #include <iostream> #include <vector> #include <queue> #include <functional> using namespace std; int main() { int n, k; cin >> n >> k; vector<int> a(n); for (int i = 0; i < n; i++) { cin >> a[i]; } sort(a.begin(), a.end()); // 对数组a进行排序 priority_queue<long long, vector<long long>, greater<long long>> pq; // 最小堆 pq.push(0); // 初始情况,前0个数的组合之和为0 for (int i = 0; i < n; i++) { long long num = pq.top(); // 取出当前队列中的最小值 pq.pop(); for (int j = i + 1; j <= n; j++) { pq.push(num + a[i]); // 将所有加和结果作为新的候选值加入队列 num += a[i]; } if (pq.size() > k) { pq.pop(); // 当队列大小超过k时,弹出最小值 } } long long sum = 0; while (!pq.empty()) { sum += pq.top(); // 求队列中所有候选值之和 pq.pop(); } cout << sum << endl; return 0; } ``` 使用优先队列的方法可以有效地找到前k小的组合之和,时间复杂度为O(nklog(k))。希望这个解法对你有所帮助!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值