poj 2274 线段树+堆

这道题可以说是数据结构题当中的一道综合题。

【题意】

给定n辆车的坐标xi，速度vi，求会发生超车事件的数量（求逆序数对），并按时间顺序输出超车事件的车对。

【题解】

第一问，经典的线段树求逆序数对，nlogn水过

第二问，比较繁。对于超车，首先必须明确一点：只有相邻的两辆车才会可能较早的发生超车。即对于三元组（i，j，k），若i能超过k，j也能超过k，则在i超过k之前，i要么超过了j，要么j超过了k，即相邻的两辆车优先。这样我们就可以使用堆维护相邻两辆车超车时间，并依次输出。

【教训】

首先，我在建堆的时候，加了个优化 ，将不能超车的排除在外，这样就会比较复杂，也没什么必要。

更重要的是，对于堆维护时，要注意状态的合法性

【代码】

#include <iostream>
using namespace std;
const int maxn=250010,oo=2147483647;


struct node
{
       int x,y;
       double k1,k2;
       node()
       {
             x=0;y=0;k1=0;k2=0;
       }
       node(int xx,int yy,double kk1,double kk2)
       {
                x=xx;y=yy;
                k1=kk1;k2=kk2;
       }           
}h[maxn*10]; 
int x[maxn],v[maxn],s[100000],next[maxn],pre[maxn];
bool vv[maxn];
int n,r,ans;
void ins(int i,int l,int r,int x)
{
     if (l==r)
     {
              s[i]++;
              return;
     }
     int mid=(l+r)/2;
     if (x<=mid) ins(i*2,l,mid,x);
     else ins(i*2+1,mid+1,r,x);
     s[i]=s[i*2]+s[i*2+1];
} 


int query(int i,int l,int r,int x,int y)
{
    if (x<=l && y>=r) return s[i];
    int mid=(l+r)/2,ans=0;
    if (x<=mid) ans+=query(i*2,l,mid,x,y);
    if (y>mid) ans+=query(i*2+1,mid+1,r,x,y);
    return ans;
}


bool operator < (node a,node b)
{
     return a.k1<b.k1 || a.k1==b.k1 && a.k2<b.k2;
}


void Swap(int i,int j)
{
     node t;
     t=h[i];h[i]=h[j];h[j]=t;
}


void up(int i)
{
     int j;
     while (i>1)
     {
           j=i/2;
           if (h[i]<h[j])
              Swap(i,j);
           else 
                break;
           i=j;
     }
}


void down(int i)
{
     int j;
     while (i*2<=r)
     {
           j=i*2;
           if (j+1<=r && h[j+1]<h[j])
              j++;
           if (h[j]<h[i])
              Swap(i,j);
           else
               break;
           i=j;
     }
}


void del()
{
     Swap(1,r);
     r--;
     down(1);
}


void add(int i,int j)
{
     double t,tt;
     if (v[i]>v[j])
        t=double(x[j]-x[i])/(v[i]-v[j]);
     else t=oo;
     tt=x[i]+v[i]*t;
     node tmp=node(i,j,t,tt);
     h[++r]=tmp;
     up(r);
}


int main()
{
    freopen("therace.i12","r",stdin);
    freopen("pou.txt","w",stdout);
    int i,j;
    scanf("%d",&n);
    next[0]=1;pre[n+1]=n;
    x[0]=0;v[0]=-oo;
    x[n+1]=oo;v[n+1]=oo;
    add(0,1);
    for (i=1;i<=n;i++)
    {
        scanf("%d%d",&x[i],&v[i]);
        ins(1,1,100,v[i]);
        ans=ans+query(1,1,100,v[i]+1,100);
        ans%=1000000;
    }
    printf("%d\n",ans);
    for (i=1;i<=n;i++)
    {
        pre[i]=i-1;next[i]=i+1;
        add(i,next[i]);
    }
    for (int ii=1;ii<=10000;ii++)
    {
        if (h[1].k1>=oo) break;
        if (pre[h[1].y]!=h[1].x || next[h[1].x]!=h[1].y) //堆状态的合法性
        {
                        --ii;
                        del();
                        continue;
        }
        printf("%d %d\n",h[1].x,next[h[1].x]);
        int xx=h[1].x,yy=next[h[1].x],pp=pre[xx],nn=next[yy];
        del();
        next[pp]=yy;
        pre[yy]=pp;next[yy]=xx;
        pre[xx]=yy;next[xx]=nn;
        pre[nn]=xx;
        add(xx,nn);
        add(pp,yy);
        add(yy,xx);
    }
    return 0;
}