「BZOJ1303」[CQOI2009] 中位数图

2014年1月6日5,7590

Description

给出1~n的一个排列，统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后，位于中间的数。

Input

第一行为两个正整数n和b ，第二行为1~n 的排列。

Output

输出一个整数，即中位数为b的连续子序列个数。

Sample Input

7 4
5 7 2 4 3 1 6

Sample Output

HINT

第三个样例解释：{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}
N<=100000

代码

找到b在数列中的位置设为point，比b大的赋值为-1，比b小的赋值为1；

然后求出sum[i,point]的值出现了几次记为lfre[sum[i,point]]++; ans += lfre[sum[i,point]]*rfre[-sum[i,point]];

由于c++数组不能是负数，所以稍微处理一下

#include<cstdio>
#include<iostream>
using namespace std;
int num[100001],sum[100001],l[200001],r[200001];
int n,b,point,ans;
int main()
{
	scanf("%d%d",&n,&b);
	for(int i=1;i<=n;i++)
    {
    	scanf("%d",&num[i]);
    	if(num[i]>b)num[i]=1;
    	else if(num[i]==b){num[i]=0;point=i;}
    	else num[i]=-1;
    }
    l[n]=1;r[n]=1;
    for(int i=point-1;i>=1;i--)
       {sum[i]=sum[i+1]+num[i];l[sum[i]+n]++;}
    for(int i=point+1;i<=n;i++)
       {sum[i]=sum[i-1]+num[i];r[sum[i]+n]++;}
    for(int i=0;i<=2*n-1;i++)ans+=l[i]*r[2*n-i];
    printf("%d",ans);
	return 0;
}

#include<cstdio>

#include<iostream>

using namespace std;

int num[100001],sum[100001],l[200001],r[200001];

int n,b,point,ans;

int main()

{

scanf("%d%d",&n,&b);

for(int i=1;i<=n;i++)

{

scanf("%d",&num[i]);

if(num[i]>b)num[i]=1;

else if(num[i]==b){num[i]=0;point=i;}

else num[i]=-1;

}

l[n]=1;r[n]=1;

for(int i=point-1;i>=1;i--)

{sum[i]=sum[i+1]+num[i];l[sum[i]+n]++;}

for(int i=point+1;i<=n;i++)

{sum[i]=sum[i-1]+num[i];r[sum[i]+n]++;}

for(int i=0;i<=2*n-1;i++)ans+=l[i]*r[2*n-i];

printf("%d",ans);

return 0;

}

« 「BZOJ1192」[HNOI2006] 鬼谷子的钱袋「JoyOI1109」幻方 »

hzwer.com _ Home | http://hzwer.com

「BZOJ1303」[CQOI2009] 中位数图

Description

Input

Output

Sample Input

Sample Output

HINT

代码

分类目录

热门文章

Myfriends

近期评论