BZOJ 1303: [CQOI2009]中位数图 问题转化_扫描_思维

将比 b 大的设成 1，比 b 小的设成 -1，开个桶左右扫描一下，乘法原理乘一乘就好了。

虽然一眼切，不过这个基于中位数的转化还是相当重要的。middle 那个主席树的题也需要该做法

Code:

#include <bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 1000000
using namespace std;
int arr[maxn],neg1[maxn],pos1[maxn],neg2[maxn],pos2[maxn],cur=0,zero1,zero2;
int main(){
    // setIO("input");
    int n,k;
    long long ans = 1;
    scanf("%d%d",&n,&k);
    for(int i = 1,a;i <= n; ++i) {
        scanf("%d",&a),arr[i]=(a<k)?-1:(a==k?0:1);
        if(a==k) cur=i;
    }
    if(cur >= 1)
    {
        for(int i = cur-1,tot=0;i>=1;--i)
        {
            tot+=arr[i];
            if(tot > 0) ++pos1[tot];
            if(tot < 0) ++neg1[-tot];
            if(tot==0) ++ans,++zero1;
        }
    }
    if(cur <= n)
    {
        for(int i = cur+1,tot=0;i<=n;++i)
        {
            tot+=arr[i];
            if(tot > 0) ++pos2[tot];
            if(tot < 0) ++neg2[-tot];
            if(tot == 0) ++ans, ++zero2;
        }
    }
    ans += (long long)zero1*zero2;
    for(int i = 1;i <= n; ++i)
    {
        ans += (long long)pos1[i]*neg2[i];
        ans += (long long)neg1[i]*pos2[i];
    }
    printf("%lld",ans);
    return 0;
}