codeforces 1005E1

最新推荐文章于 2024-10-06 21:54:42 发布

原创最新推荐文章于 2024-10-06 21:54:42 发布 · 208 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ACM #codeforce

cf 专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种求解特定位置中位数的算法，通过预处理子串，统计元素个数并排序，实现高效匹配。算法适用于处理排列数据，通过左右子串的value计算，快速定位中位数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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;
}