hdu 4302 Holedox Eating(优先队列)

最新推荐文章于 2021-04-01 17:21:02 发布

qian99

最新推荐文章于 2021-04-01 17:21:02 发布

阅读量931

点赞数

分类专栏：数据结构文章标签：数据结构

本文链接：https://blog.csdn.net/qian99/article/details/12915861

版权

数据结构专栏收录该内容

91 篇文章 0 订阅

订阅专栏

题意：有一个小动物在长L的数轴上，最开始这只小动物在0点，每个时候，有可能有两种事件，第一种是数轴上某一点会出现一份食物，另一种是小动物想找吃的，这种情况小动物会找最近的一个食物，然后过去，如果两边的食物距离相同，那么就按最后一次移动的方向去，如果两边都没有食物，就呆在原地。

思路：刚开始想用线段树做来着，但是想了想感觉好麻烦，后来想这题其实就是从左边找个最大值，从右边找个最小值嘛，那就可以用两个优先队列搞了。还有一点就是注意一下当前位置就有食物的情况就行了。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
using namespace std;
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int t,L,n,tcase=0;
    scanf("%d",&t);
    while(t--)
    {
        tcase++;
        priority_queue<int>ql;
        priority_queue<int,vector<int>,greater<int> >qr;
        scanf("%d%d",&L,&n);
        int now=0,scnt=0;
        int tp,v,lmove=-1;
        int sum=0;
        while(n--)
        {
            scanf("%d",&tp);
            if(tp)
            {
                if(scnt) scnt--;
                else
                {
                    int lp=-1,rp=-1;
                    if(!ql.empty()) lp=ql.top();
                    if(!qr.empty()) rp=qr.top();
                    if(lp==-1&&rp==-1) continue;
                    if(lp==-1) lmove=1;
                    else if(rp==-1) lmove=0;
                    else if(now-lp<rp-now) lmove=0;
                    else if(now-lp>rp-now) lmove=1;
                    if(lmove)
                    {
                        sum+=(rp-now);
                        now=rp;
                        scnt=0;
                        qr.pop();
                        while(!qr.empty()&&(qr.top()==rp))
                        {scnt++;qr.pop();}
                    }
                    else
                    {
                        sum+=(now-lp);
                        now=lp;
                        scnt=0;
                        ql.pop();
                        while(!ql.empty()&&(ql.top()==lp))
                        {scnt++;ql.pop();}
                    }
                }
            }
            else
            {
                scanf("%d",&v);
                if(v>now) qr.push(v);
                else if(v<now) ql.push(v);
                else scnt++;
            }
        }
        printf("Case %d: %d\n",tcase,sum);
    }
    return 0;
}