一道2017年百度笔试编程题

来源:互联网 发布:上海云计算公司排名 编辑:程序博客网 时间:2024/05/17 07:32
前K大数时间限制:C/C++语言 2000MS;其他语言 4000MS
内存限制:C/C++语言 65536KB;其他语言 589824KB题目描述:
小M即将过生日了,有n个人将要送她礼物。她对每个生日礼物都有一个喜好值,值越大表示越喜欢。她想从中选出K个她最喜欢的礼物。但是由于所有的礼物并不是同时送到的,所以她想知道,每收到一个礼物后,她最喜欢的K个礼物的喜好值之和是多少。
输入
第一行包含两个整数n,K。10≤n≤105 ,2≤K≤n。
第二行包含n个整数ai,表示小M对第i个收到的物品的喜好值。1≤ai≤10000
输出
输出n-K+1个整数,表示小M新收到一个礼物后最喜欢的个礼物的喜好值之和。


样例输入
14 4
1 2 3 4 5 6 7 8 9 10 1 1 1 1
样例输出

10 14 18 22 26 30 34 34 34 34 34



太弱鸡,知道要用堆排序,可是那一堆 make_heap push_heap sort_heap push_back pop_back back彻底把我整懵逼了,完全不知道哪个是哪个了,晕头转向,最后没招,用qsort吧,自己不会写快排,毕竟,,,


仅供参考,写的太烂,大神勿喷

#include <iostream>#include <vector>#include<algorithm>using namespace std;void show(vector<int> hehe){    for(unsigned int i=0;i!=hehe.size();i++)cout<<hehe[i]<<" ";    cout<<endl;}int cmp(const void *a, const void *b){    return *(int *)a - *(int *)b; //进行升序排序,降序a,b位置互换}int main(){    //cout << "Hello world!" << endl;    const int MAX=1024;    int n,K;    int a[MAX]={-1};    cin >> n >> K;    for(int i=0;i<n;i++)    {        cin >> a[i];    }    vector<int> ivec(a,a+K);    //show(ivec);    //第一个元素为最小值    //make_heap(ivec.begin(),ivec.end(),greater<int>());    //show(ivec);    //从大到小排序    //sort_heap(ivec.begin(),ivec.end(),greater<int>());    qsort(a,K,sizeof(int),cmp);//    for(int i=0;i<K;i++)//    {//        cout << "is "<<a[i] << endl;//    }    //show(ivec);    //cout << "pop heap..."<<ivec.back()<<endl;    //cout<< "111"<<endl;    //show(ivec);    int minnum=a[0];    //int minnum=ivec.back();    //cout<< "222"<<endl;    //show(ivec);    //pop_heap(ivec.begin(),ivec.end());    //14 4//1 2 3 4 5 6 7 8 9 10 1 1 1 1    int sum=0;    for(int i=0;i<K;i++)    {        sum+=a[i];//////        cout << "sum is "<<sum<<endl;////        cout << "minnum is "<<minnum <<" a["<< i<<"] is " << a[i]<<endl;////        if(minnum>a[i])////        {////            cout << "minnum is "<<minnum<<endl;////            minnum=a[i];////        }    }    cout << sum << " ";    for(int i=K;i<n;i++)    {        if(a[i]>minnum)        {            //cout << "minnum is "<<minnum<<endl;            sum-=minnum;            //sort_heap(ivec.begin(),ivec.end(),greater<int>());            //cout << "pop heap..."<<ivec.back()<<endl;            //cout << "size: "<<ivec.size()<<endl;            //cout << "pop_back before..."<<ivec.back()<<endl;            //ivec.pop_back();            //cout << "pop_back after..."<<ivec.back()<<endl;            //cout << "size: "<<ivec.size()<<endl;            //show(ivec);            //show(ivec);            //cout << "size: "<<ivec.size()<<endl;            //make_heap(ivec.begin(),ivec.end(),greater<int>());//这里可以用make_heap替换。            //push_heap(ivec.begin(),ivec.end());            qsort(a,i,sizeof(int),cmp);            minnum=a[i-K+1];            //cout << "push_back:"<<a[i]<<endl;            //cout << "pop_back before...222..."<<ivec.back()<<endl;            //ivec.push_back(a[i]);            //cout << "pop_back after...222..."<<ivec.back()<<endl;            //show(ivec);            //ivec.pop_back();            //show(ivec);            sum+=a[i];            //cout << "minnum is "<<minnum<<endl;            //cout << sum << " ";        }        cout << sum<<" ";    }    cout << endl;    return 0;}