nowcoder172A 中位数 (二分答案)

二分一下答案,假设是x。

我们把大于x的看成1,小于x的看成-1,等于x的看成0

那某个区间的和如果是正的,就说明这个区间中位数大于x;如果是0,就等于x;如果是负的,就小于x;

这样的话,做一个前缀和,然后维护一下满足长度>=len的最小值,然后减一减,看看是不是>=0就可以了。

(我自己写的话我自己都看不懂...)

 1 #include<bits/stdc++.h>
 2 #define pa pair<int,int>
 3 #define ll long long
 4 using namespace std;
 5 const int maxn=100010;
 6 
 7 ll rd(){
 8     ll x=0;char c=getchar();int neg=1;
 9     while(c<'0'||c>'9'){if(c=='-') neg=-1;c=getchar();}
10     while(c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
11     return x*neg;
12 }
13 
14 int N,num[maxn],val[maxn],mlen;
15 int sum[maxn];
16 
17 inline bool judge(int x){
18     for(int i=1;i<=N;i++) sum[i]=sum[i-1]+(num[i]<x?-1:num[i]>x);
19     int mi=0x3f3f3f3f;bool b=0;
20     for(int i=mlen+1;i<=N;i++){
21         mi=min(mi,sum[i-mlen]);
22         b|=(sum[i]>=mi);
23     }return b;
24 }
25 
26 int main(){
27     int i,j,k;
28     N=rd(),mlen=rd();
29     for(i=1;i<=N;i++) val[i]=num[i]=rd();
30     sort(val+1,val+N+1);
31     int l=1,r=N;
32     while(l<=r){
33         int m=l+r>>1;
34         if(judge(val[m])) l=m+1;
35         else r=m-1;
36     }printf("%d\n",val[l-1]);
37     return 0;
38 }

猜你喜欢

转载自www.cnblogs.com/Ressed/p/9623531.html