BZOJ 1303
【题目链接】
http://www.lydsy.com/JudgeOnline/problem.php?id=1303
【解题报告】
思路比较巧妙,重新定义数组,比b大的数设为-1,比b小的数设为1。
为什么可以这么想呢?对于中位数而言,我们只关心这个连续序列里有多少个元素比他大,而不关心比他大多少,即比他大的数无论怎么排列都没有关系。
也即,数列里的数只有三个性质:大于b,小于b,等于b。
那么我们使用-1,0,1即可表示数列里的数的性质。
为什么要维护这个性质呢?
因为我们需要在O(N)时间内求得以b为中位数的区间有多少个。
也即,对一个区间,我们需要O(1)查询b是不是中位数。
这里的思路最关键。我们设pos为b所在的位置,那么在区间(l,r)内,
我们知道(pos+1,r)的sum。
sum=0,大于b和等于b的数一样多
sum>0,大于b的数比小于b的数多sum个
sum<0,同理
那么如果sum(l,pos-1)+sum(pos+1,r)=0,是不是说,在(l,r)内大于b和小于b的数是一样多的?
所以我们只需要跑一遍前缀和即可。
还有一个细节,需要对数组下标进行平移。读者可以在编写代码的过程中自行分析和设计。
【参考代码】
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int N,b;
int hashL[200000+50],hashR[200000+50];
int sum[100000+50],a[100000+50];
int main()
{
while( ~scanf("%d%d",&N,&b) )
{
int pos;
memset( sum,0,sizeof sum );
memset( hashL,0,sizeof hashL );
memset( hashR,0,sizeof hashR );
for( int i=1; i<=N; i++ )
{
scanf( "%d",&a[i] );
if( a[i]>b ) a[i]=1;
else if( a[i]==b ){ a[i]=0; pos=i; }
else a[i]=-1;
}
for( int i=pos-1; i>=1; i-- )
{
sum[i]=sum[i+1]+a[i];
hashL[sum[i]+N]++;
}
for( int i=pos+1; i<=N; i++ )
{
sum[i]=sum[i-1]+a[i];
hashR[sum[i]+N]++;
}
long long ans=0;
hashL[N]+=1;
hashR[N]+=1;
for( int i=1; i<=2*N; i++ )
{
ans+=(long long)hashL[i]*(long long)hashR[N*2-i];
}
cout<<ans<<endl;
// for( int i=1; i<=N; i++ ) cout<<a[i]<<endl;
}
return 0;
}
根据这道题,有一道变形的题目:
https://biancheng.love/problem/214/index
–2015 BCPC problem B
原理和本题是一样的。
但是需要考虑如果有相同的元素在序列里,应该怎么处理?
请读者自行思考。
【参考代码】
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long LL;
const int maxn=5000+50;
int N;
LL ans;
int A[maxn],B[maxn];
void change( int num, int pos )
{
for( int i=1; i<=N; i++ )
{
if( A[i]<num || ( A[i]==num && i<pos ) )B[i]=-1;
else if( A[i]>num || ( A[i]==num && i>pos ) ) B[i]=1;
else B[i]=0;
}
}
int L[maxn*2],R[maxn*2],sum[maxn];
void getsum( int pos )
{
memset( L,0,sizeof L );
memset( R,0,sizeof R );
memset( sum,0,sizeof sum );
L[N]=R[N]=1;
for( int i=pos-1; i>=1; i-- )
{
sum[i]=sum[i+1]+B[i];
L[ sum[i]+N ]++;
}
for( int i=pos+1; i<=N; i++ )
{
sum[i]=sum[i-1]+B[i];
R[ sum[i]+N ]++;
}
}
void solve( int pos )
{
for( int i=1; i<=2*N; i++ )
{
ans+=(LL)L[i]*(LL)R[ 2*N-i ]*(LL)A[pos];
ans+=(LL)L[i]* (LL)R[ 2*N-i+1 ]*(LL)A[pos];
}
}
int main()
{
while( ~scanf( "%d",&N ) )
{
for( int i=1; i<=N; i++ ) scanf( "%d", &A[i] );
ans=0;
for( int i=1; i<=N; i++ )
{
change( A[i], i );
getsum( i );
solve( i );
}
printf( "%lld\n",ans );
}
return 0;
}