BZoj 1293 生日礼物(小顶堆)

传送门:http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=27400


Anayse:小顶堆;

1 1 5     - >    1 3 5     ->    3 5 7    ->     5 7 8

pop(1)    p(1)      p(3)       最优解。

每次剔除堆顶,然后换push与堆顶相同种类的珠子,由于单调,,距离会被缩短,于是找出最优解。详见上式。

push()  log(n)

so,复杂度 nlog(n).



Code:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<string>
#include<queue>
#include<deque>
#include<stack>
#include<map>
#include<set>
#define INF 0x7fffffff
#define SUP 0x80000000
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;

typedef long long LL;
const int N=100007;


int **a;
int nu[100];


struct node{
    int id,pos;
    node(int a,int b):id(a),pos(b){};
    bool operator <(node a) const
    {
        return pos<a.pos;
    }
};
vector<node> aa;

int main()
{
    int n,k;
    while(scanf("%d%d",&n,&k)==2)
    {
        mem(nu,0);
        priority_queue<int,vector<int>,less<int> > que;
        a=new int *[k+1];
        int num;
        for(int i=1;i<=k;i++)
        {
            scanf("%d",&num);
            a[i]=new int[num+1];
            nu[i]=a[i][0]=num;
            for(int j=1;j<=num;j++)
                scanf("%d",&a[i][j]);
        }

        int cnt=k;
        for(int i=1;i<=k;i++)  //构造初始堆
        {
            aa.push_back(node(i,a[i][1]));   
            nu[i]=2;
        }
        int ans=INF;
        while(cnt<=n)   //依次剔除堆顶
        {
            sort(aa.begin(),aa.end());
            ans=min(ans,aa[k-1].pos-aa[0].pos);
            int ii=aa[0].id;
            if(nu[ii]>a[ii][0]){
                break;
            }
            aa[0]=node(ii,a[ii][nu[ii]++]);
            cnt++;
        }

        printf("%d\n",ans);

    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值