[bzoj1303]中位数图

由于是排列,因此b一定只出现了一次,找到出现的位置并向左右扩展
考虑如何判定是否满足条件,当且仅当$[左边比b小的数ls]+[右边比b小的数rs]=[左边比b大的数lb]+[右边比b大的数rb]$,暴力枚举+线段树复杂度为$o(n^2logn)$
变形上式,得到$ls-lb=rb-rs$,对两边分别处理后对应方案相乘求和即为答案

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 int n,m,a[100005],s1[200005],s2[200005];
 4 long long ans;
 5 int main(){
 6     scanf("%d%d",&n,&m);
 7     for(int i=1;i<=n;i++)scanf("%d",&a[i]);
 8     for(int i=1;i<=n;i++)
 9         if (a[i]==m){
10             int t=n;
11             s1[n]=s2[n]=1;
12             for(int j=i-1;j;j--)s1[t+=2*(a[j]<m)-1]++;
13             t=n;
14             for(int j=i+1;j<=n;j++)s2[t+=2*(a[j]>m)-1]++;
15             for(int j=1;j<=2*n;j++)ans+=1LL*s1[j]*s2[j];
16             printf("%lld",ans);
17             return 0;
18         }
19 }
View Code

 

posted @ 2019-11-08 16:19  PYWBKTDA  阅读(275)  评论(0编辑  收藏  举报