BZOJ#1303中位数图
[CQOI2009]中位数图
因为是排列,不会出现相同的数字,把比b小的赋值为1,大于b的赋值为-1,则要是b为中位数必然这一段的和为0,(注意b的位置)
代码:
const int N=200010;
int a[N];
int s[N];
void solve(int Case) {
int n,b;
cin>>n>>b;
for(int i=1;i<=n;i++) cin>>a[i];
int p=-1;
for(int i=1;i<=n;i++){
if(a[i]<b) s[i]=1;
else if(a[i]>b) s[i]=-1;
if(a[i]==b) p=i;
}
for(int i=1;i<=n;i++) s[i]+=s[i-1];
map<int,int> mp;
mp[0]++;
int res=0;
for(int i=1;i<=n;i++){
if(i>=p){
res+=mp[s[i]];
}
if(i<p) mp[s[i]]++;
}
cout<<res<<nline;
}