和中位数有关的构造

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;
}