51nod 1682

原题链接

显然由题意描述可知（所有数皆是独一无二的），所有符合条件的区间皆为奇数长度的区间。

对于一个区间（l,r），定义up为区间内大于x的数的个数，down为小于x的数的个数，若x为中位数则有up(x)=down(x)，cnt[x]表示up-down=x的区间有多少个。

假设x在区间（l,r）中的位置为pos，x为中位数则要满足在（l,pos-1）中up-down=在（pos+1,r）中down-up。

每次枚举一个数a[i]，统计在区间(l,i)内up和down的值并更新cnt[up-down]的值，并不断向左扩展l，右区间也同样操作。

#include<map>
#include<set>
#include<list>
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<cstdio>
#include<string>
#include<bitset>
#include<vector>
#include<utility>
#include<numeric>
#include<cstring>
#include<iterator>
#include<iostream>
#include<algorithm>
#include<functional>
using namespace std;
#define fi first
#define se second
#define MP make_pair
#define ll long long
#define PR pair<ll,ll>
#define LS (root<<1)
#define RS ((root<<1)|1)
#define LSON LS,l,mid
#define RSON RS,(mid+1),r
#define MID mid=((l+r)/2)
#define mm(a,b) memset(a,b,sizeof(a))
#define rep(a,b,c) for(int a=b;a<c;a++)
#define INF 0x3f3f3f3f3f3f3f3f
#define eps 1e-9
#define Pi acos(-1.0)
#define MAXN 8005
#define MAXE 3000050
#define Inf 9999999
#define debug puts("**************");
template<class T> T f_max(T a, T b){ return a > b ? a : b; }
template<class T> T f_min(T a, T b){ return a < b ? a : b; }
template<class T> T f_abs(T a){ return a > 0 ? a : -a; }
template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template<class T> T lcm(T a, T b){ return a / gcd(a, b)*b; }
const int mod=1000007;
int n;
int a[MAXN];
int cnt[2*MAXN];
int main(){
    int T, cas=1;
    int ans=0;

    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
//            b[i]=a[i];
        }
        for(int i=1;i<=n;i++){
            ans=0; mm(cnt,0);
            int up, down;
            up=down=0;
            cnt[8000]++;
            for(int j=i-1;j>=1;j--){ //向左扩展
                if(a[j]>a[i]) up++;
                if(a[j]<a[i]) down++;
                cnt[down-up+8000]++;
            }
            up=down=0;
            for(int j=i;j<=n;j++){ //向右扩展
                if(a[j]>a[i]) up++;
                if(a[j]<a[i]) down++;
                ans+=cnt[up-down+8000];
            }
            printf("%d%c",ans,i==n?'\n':' ');
        }
    }

    return 0;
}