codeforces 1005E1

E1. Median on Segments (Permutations Edition)

time limit per test：3 seconds    memory limit per test：256 megabytes

 

对待查位置左右子串预处理，统计大于/小于该数的元素个数并作差，得到value。

分别统计左右两侧各value的出现次数，并分别按照value由小到大、由大到小进行排序。如果l[a]+r[b]>1，则l[a+1]+r[b]一定大于1，所以每次匹配不需要从头开始。

左右两侧的位置匹配+单独左侧+单独右侧+数字本身位置为最终输出结果。

 

#include <bits/stdc++.h>
using namespace std;

struct inf{
    long long times=0;
    long long value=0;
}l[200005],r[200005];

long long input[200005];

bool cmpl(inf a,inf b){
    return a.value<b.value;
}

bool cmpr(inf a,inf b){
    return a.value>b.value;
}

int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=0;i<n;i++)
        cin>>input[i];
    int work=find(input,input+n,m)-input;
    //cout<<work<<endl;
    input[work]=0,input[n]=2000000;
    for(int i=work-1;i>=0;i--){
        if(input[i]>m)
            input[i]=input[i+1]+1; //大于0表示比m大的个数
        else
            input[i]=input[i+1]-1;
    }
    for(int i=work+1;i<n;i++){
        if(input[i]>m)
            input[i]=input[i-1]+1; //大于0表示比m大的个数
        else
            input[i]=input[i-1]-1;
    }
    sort(input,input+work);
    sort(input+work+1,input+n);
    /*for(int i=0;i<n;i++){
        cout<<input[i]<<" ";
    }*/
    input[work]=2000000;
    int lnum=0;
    for(int i=0;i<work;i++){
        l[lnum].times++;
        if(input[i+1]!=input[i]){
            l[lnum].value=input[i];
            lnum++;
        }
    }
    int rnum=0;
    for(int i=work+1;i<n;i++){
        r[rnum].times++;
        if(input[i+1]!=input[i]){
            r[rnum].value=input[i];
            rnum++;
        }
    }
    sort(l,l+lnum,cmpl);
    sort(r,r+rnum,cmpr);
    //cout<<lnum<<" "<<rnum;
    /*for(int i=0;i<lnum;i++){
        cout<<l[i].value<<" ";
    }
    cout<<endl;
    for(int i=0;i<rnum;i++){
        cout<<r[i].value<<" ";
    }*/
    int lwork=0,rwork=0;
    long long out=0;
    for(;lwork<lnum;lwork++){
        for(;r[rwork].value+l[lwork].value>1 && rwork<rnum;rwork++);
        if(rwork==rnum)
            break;
        if(r[rwork].value+l[lwork].value==1 || r[rwork].value+l[lwork].value==0)
            out+=r[rwork].times*l[lwork].times;
        //cout<<out<<endl;
        if(rwork+1==rnum)
            continue;
        if(r[rwork+1].value+l[lwork].value==0 || r[rwork+1].value+l[lwork].value==1)
            out+=r[rwork+1].times*l[lwork].times;
        //cout<<out<<endl;
    }
    long long ltemp=0;
    for(int i=0;i<lnum;i++){
        if(l[i].value==0 || l[i].value==1)
            ltemp+=l[i].times;
    }
    long long rtemp=0;
    for(int i=0;i<rnum;i++){
        if(r[i].value==0 || r[i].value==1)
            rtemp+=r[i].times;
    }
    cout<<out+1+ltemp+rtemp;
    return 0;
}