HDU 6168 Numbers 优先队列

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=6168
题意:有一个长度为n的序列a1……an,根据a序列生成了一个b序列,b[i] = a[i]+aj,然后有一个人把a,b序列按随机顺序混合了起来,现在问你初始的a序列是什么 ?

思路:用已确定的a[]数组来确定b[]数组的元素,从已知序列中删掉,再确定新的a[]数组元素,重复步骤。

  模拟一下(假定输入数组为ss[]数组):
  样例二: 1 2 3 3 4 4 5 5 5 6 6 6 7 7 7 8 8 9 9 10 11
  首先,ss[]数组最小的两个数一定为a[]数组的前两个元素,我们以此为基础向后更新,因为直接排序会超时,这里我用是优先队列,即将ss[]数组入队num队列(队列sum是a[]数组和b[]数组的混合),
  此时初始队列为:1 2 3 3 4 4 5 5 5 6 6 6 7 7 7 8 8 9 9 10 11;

  先将a1和a2出队,而a1+a2=1+2=3;所以我们num队列中的一个3出队(由a1和a2组成的是b[]数组中的一个元素),num队列中剩下的3一定是a[]数组中的元素(此时我们确定a[]数组的元素组合只能组合出一个3),出队;
  此时num队列变为4 4 5 5 5 6 6 6 7 7 7 8 8 9 9 10 11;

  现在我们已知的a[]数组元素有1,2,3;他们能组成的b[]数组元素为4,5(这里只考虑包含新元素的情况);我们将确认的b[]数组元素放入另一个优先队列s;
  队列s:4,5;(这里解释一下,队列s所存的元素就是b[]数组中的元素。我们用的是优先队列,每次最小的元素出队,用队列s去更新队列sum,找出a[]数组的元素。)

  num队列中的4出队,此时队列中还有一个4,则4就是我们确定的a[]数组中的新元素(同理此时我们确定a[]数组的元素组合只能组合出一个4),现在我们已知的a[]数组元素有1,2,3,4;之后num队列中的5出队,此时队列s为空,以确认的新元素4更新队列s:5,6,7;重复以上步骤,最后得出所求数组a[]:1,2,3,4,5,6;
  可能我写的有点混乱,但事实上代码是非常好理解的,调试一遍代码就清楚了(那我写这个有什么用?)。
  Ps:感觉自己菜炸了 O_O !

#include<bits/stdc++.h>
using namespace std;

const int maxn= 125500;
priority_queue<int,vector<int>,greater<int> > num;
priority_queue<int,vector<int>,greater<int> > s;
int a[maxn];

int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        int x;
        for(int i=0; i<n; i++)
        {
            scanf("%d",&x);
            num.push(x);
        }
        ///前两个数
        int tmp=0,tmp1;
        a[tmp++]=num.top();
        num.pop();
        a[tmp++]=num.top();
        num.pop();

        s.push(a[0]+a[1]);//和
        int l=0;
        while(!num.empty())
        {
            int temp=num.top();
            if(s.empty())
            {
                int w=num.top();
                a[tmp++]=w;
                for(int i=0; i<tmp-1; i++)
                {
                    s.push(a[i]+w);
                }
                num.pop();
                continue;
            }
            int temp1=s.top();
            if(temp==temp1)
            {
                num.pop();
                s.pop();
            }
            else
            {
                int w=num.top();
                a[tmp++]=w;
                for(int i=0; i<tmp-1; i++)
                {
                    s.push(a[i]+w);
                }
                num.pop();
            }
        }
        printf("%d\n",tmp);
        for(int i=0; i<tmp; i++)
        {
            if(i!=0)
                printf(" ");
            printf("%d",a[i]);
        }
        printf("\n");
    }
}
  • 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、付费专栏及课程。

余额充值