首页 > 代码库 > BZOJ 1303: [CQOI2009]中位数图 【水题】
BZOJ 1303: [CQOI2009]中位数图 【水题】
给出1~n的一个排列,统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后,位于中间的数。
Input
第一行为两个正整数n和b ,第二行为1~n 的排列。
Output
输出一个整数,即中位数为b的连续子序列个数。
Sample Input
7 4
5 7 2 4 3 1 6
Sample Output
4
HINT
第三个样例解释:{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}
N<=100000
思路:记录中间那个数左边右边 比它大和比它小的个数,然后乘法原理搞一下
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <queue>
#define maxn 500000
#define com 100005
using namespace std;
int righ[maxn],lef[maxn],a[maxn];
int main()
{
int n,b,idx;
long long ans=0;
scanf("%d%d",&n,&b);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
if(a[i]==b)idx=i;
}
int sum=0;lef[com]=righ[com]=1;
for(int i=idx-1;i>=1;i--)
{
sum+=(a[i]>b)?1:-1;
lef[sum+com]++;
}
sum=0;
for(int i=idx+1;i<=n;i++)
{
sum+=(a[i]<b)?1:-1;
righ[sum+com]++;
}
for(int i=com-n;i<=com+n;i++)ans+=lef[i]*righ[i];
cout<<ans<<endl;
return 0;
}
BZOJ 1303: [CQOI2009]中位数图 【水题】