题意:
![在这里插入图片描述](https://i-blog.csdnimg.cn/direct/843a4303625c4c378aebc00f0bad688d.png)
解法:
直接求中位数=m的方案数不好求
问题可以转换为中位数>=m的方案数-中位数>=m+1的方案数
感觉和一些数位dp求[l,r]内方案数的想法有异曲同工之妙.
如何计算中位数>=mid的方案数:
令b[i]=(a[i]>=mid)?1:-1
中位数>=mid的区间, 一定满足区间和>0
那么问题可以转换为有多少个子区间,满足区间和>0
即sum[i]-sum[l]>0, sum[i]>sum[l].
对于每个i,计算满足条件的l的数量即可.
可以用前缀和+树状数组解决.
这题由于数值是+-1,因此前缀和的变化不会超过1,这种情况可以不用树状数组,直接维护有多少个满足条件的l.
Code:
#include <bits/stdc++.h>
using namespace std;
#define X first
#define Y second
#define int long long
#define PI pair<int, int>
const int maxm=2e6+5;
const int mod=1e9+7;
int n,m;
int a[maxm];
int b[maxm];
int cal(int mid){
for(int i=1;i<=n;i++){
if(a[i]>=mid)b[i]=1;
else b[i]=-1;
}
map<int,int>mp;
int ans=0;
int sum=0;
int cnt=0;
for(int i=1;i<=n;i++){
mp[sum]++;
if(b[i]==1){
cnt+=mp[sum];
ans+=cnt;
sum++;
} else {
sum--;
cnt-=mp[sum];
ans+=cnt;
}
}
return ans;
}
void solve() {
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i];
}
int ans=cal(m)-cal(m+1);
cout<<ans<<endl;
}
signed main() {
ios::sync_with_stdio(0);
cin.tie(0);
#ifndef ONLINE_JUDGE
freopen("../in.txt", "r", stdin);
freopen("../out.txt", "w", stdout);
#endif
#ifdef MULTI_CASE
int T;
cin >> T;
while (T--)
#endif
solve();
return 0;
}