BZOJ 3314 [Usaco2013 Nov]Crowded Cows：单调队列

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define MAX_N 50005

using namespace std;

struct Data
{
    int x;
    int h;
    Data(int _x,int _h)
    {
        x=_x;
        h=_h;
    }
    Data(){}
    friend bool operator < (const Data &a,const Data &b)
    {
        return a.x<b.x;
    }
};

int n,d;
int head;
int tail;
int ans=0;
int q[MAX_N];
int cnt[MAX_N];
Data dat[MAX_N];

void read()
{
    cin>>n>>d;
    for(int i=0;i<n;i++)
    {
        cin>>dat[i].x>>dat[i].h;
    }
}

void solve()
{
    sort(dat,dat+n);
    memset(cnt,0,sizeof(cnt));
    head=0;
    tail=0;
    for(int i=0;i<n;i++)
    {
        while(head<tail && dat[i].x-dat[q[head]].x>d) head++;
        while(head<tail && dat[q[tail-1]].h<=dat[i].h) tail--;
        if(head<tail && dat[q[head]].h>=dat[i].h*2) cnt[i]++;
        q[tail++]=i;
    }
    head=0;
    tail=0;
    for(int i=n-1;i>=0;i--)
    {
        while(head<tail && dat[q[head]].x-dat[i].x>d) head++;
        while(head<tail && dat[q[tail-1]].h<=dat[i].h) tail--;
        if(head<tail && dat[q[head]].h>=dat[i].h*2) cnt[i]++;
        q[tail++]=i;
    }
    for(int i=0;i<n;i++)
    {
        if(cnt[i]==2) ans++;
    }
}

void print()
{
    cout<<ans<<endl;
}

int main()
{
    read();
    solve();
    print();
}