JZOJ5238【GDOI模拟】的士碰撞

Description

n辆车在一条数轴上，车的编号为1到n。编号为i的车坐标为a[i]，初始方向为dir[i]（左或右），初始位置两两不同。每辆车每个时刻行走距离为1。两辆车相碰时，会调转方向，继续行走，掉头不消耗时间。现在车子开始朝其方向行驶，同一个坐标允许有多辆车。现在有q个询问，给出 t,i，询问过了t时刻后，编号为i的车的坐标的绝对值。

Solution

首先我们可以知道，相对顺序是不会变的，就是从左到右第i个，最后还是从左到右第i个。然后我们知道相撞只是符号交换一下，最后这个符号的位置可能会移动T(和方向有关)
那么我们可以二分一下最后的位置mid，然后求有多少个数是小于mid的，知道这个个数为i-1，因为相对顺序不变。求多少个数小于mid，对于向左向右的分开二分，向左的找a[i]-T

Code

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=1e5+7,inf=0x7fffffff;
typedef long long ll;
ll i,j,k,t,n,m,ans,x,y,l1,r1,mid1;
struct node{
    ll a,b,c;
}a[maxn];
ll l,r,mid;
bool cmp(node x,node y){return x.a<y.a;}
int b[maxn],c[maxn],w[maxn];
int main(){
    freopen("collision.in","r",stdin);
    freopen("collision.out","w",stdout);
    scanf("%lld%lld",&n,&m);
    fo(i,1,n)scanf("%lld%lld",&a[i].a,&a[i].b),a[i].c=i;
    sort(a+1,a+1+n,cmp);fo(i,1,n)w[a[i].c]=i;
    fo(i,1,n)if(!a[i].b)b[++b[0]]=a[i].a;else c[++c[0]]=a[i].a;
    fo(i,1,m){
        scanf("%lld%lld",&x,&y);
        l=-inf,r=inf;
        while(l<r){
            mid=(l+r+1)/2;t=0;
            l1=1,r1=b[0];
            while(l1<r1){
                mid1=(l1+r1+1)/2;
                if(mid<b[mid1]-x)r1=mid1-1;else l1=mid1;
            }
            if(mid<b[l1]-x)l1--;
            t+=l1;
            l1=1,r1=c[0];
            while(l1<r1){
                mid1=(l1+r1+1)/2;
                if(mid<c[mid1]+x)r1=mid1-1;else l1=mid1;
            }
            if(mid<c[l1]+x)l1--;
            t+=l1;
            if(t>=w[y])r=mid-1;else l=mid;
        }
        printf("%lld\n",abs(l+1));
    }
}