bzoj1303: [CQOI2009]中位数图

链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1303

题意：中文题。。

分析：这题还是挺水的，既然是一个1~n的排列，那么我们自然就可以把大于b的数变为1，小于b的数变为-1，那么b自然就是0咯，然后我们就是在数列中0的位置的左边和右边匹配相同的前缀和就可以啦。

代码：

#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<math.h>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=100010;
const int MAX=151;
const int MOD=1000007;
const int MOD1=100000007;
const int MOD2=100000009;
const int INF=2100000000;
const double EPS=0.00000001;
typedef long long ll;
typedef unsigned long long ull;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int a[N],sum[N],l[2*N],r[2*N];
int main()
{
    int i,j,b,n,mid;
    ll ans=0;
    scanf("%d%d", &n, &b);
    for (i=1;i<=n;i++) {
        scanf("%d", &a[i]);
        if (a[i]==b) a[i]=0;
        else if (a[i]>b) a[i]=1;
            else a[i]=-1;
    }
    sum[0]=n;
    for (i=1;i<=n;i++) {
        sum[i]=sum[i-1]+a[i];
        if (a[i]==0) mid=sum[i];
    }
    for (i=1;i<=n;i++)
    if (a[i]==0) break ;
    else l[sum[i]]++;
    for (j=i+1;j<=n;j++) r[sum[j]]++;
    l[n]++;r[mid]++;
    for (i=0;i<=2*n;i++) ans+=(ll)l[i]*r[i];
    printf("%lld\n", ans);
    return 0;
}


/*
7 4
5 7 2 4 3 1 6

*/