1. 程式人生 > >BZOJ 1303: [CQOI2009]中位數圖(思路題)

BZOJ 1303: [CQOI2009]中位數圖(思路題)

傳送門

解題思路

  比較好想的思路題。首先肯定要把原序列轉化一下,大於\(k\)的變成\(1\),小於\(k\)的變成\(-1\),然後求一個字首和,還要用\(cnt[]\)記錄一下字首和每個數出現了幾次,然後統計答案的時候從\(1\)迴圈到\(k\),每次轉化為字首和相減即可。

程式碼

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;
const int MAXN = 100005;
typedef long long LL;

inline int rd(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {f=ch=='-'?0:1,ch=getchar();}
    while(isdigit(ch)) {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return f?x:-x;
}

int n,m,a[MAXN],p;
int cnt[MAXN<<1],tag;
LL ans;

int main(){
    n=rd(),m=rd();
    for(int i=1;i<=n;i++){
        a[i]=rd();
        if(a[i]>m) a[i]=1;
        else if(a[i]==m) a[i]=0,p=i;
        else a[i]=-1;
        a[i]+=a[i-1];
        if(p) cnt[a[i]+100000]++;
    }
    ans+=cnt[100000];
    for(int i=1;i<p;i++) ans+=cnt[a[i]+100000];
    printf("%lld\n",ans);
    return 0;
}