牛客ACM-[USACO 2011 Nov G]Above the Median

最新推荐文章于 2024-09-12 19:02:20 发布

花崽oyf

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量166

点赞数 3

分类专栏：牛客-树状数组文章标签：算法 c++ 图论

本文链接：https://blog.csdn.net/ouyunfeng/article/details/126554563

版权

牛客-树状数组专栏收录该内容

3 篇文章 0 订阅

订阅专栏

[USACO 2011 Nov G]Above the Median

题目描述

Farmer John has lined up his N (1 <= N <= 100,000) cows in a row to measure their heights; cow i has height Hi(1<=Hi<=1,000,000,000)H_i (1 <= H_i <= 1,000,000,000)Hi(1<=Hi<=1,000,000,000) nanometers–FJ believes in precise measurements! He wants to take a picture of some contiguous subsequence of the cows to submit to a bovine photography contest at the county fair.
The fair has a very strange rule about all submitted photos: a photograph is only valid to submit if it depicts a group of cows whose median height is at least a certain threshold X (1 <= X <= 1,000,000,000).
For purposes of this problem, we define the median of an array A[0…K] to be A[ceiling(K/2)] after A is sorted, where ceiling(K/2) gives K/2 rounded up to the nearest integer (or K/2 itself, it K/2 is an integer to begin with). For example the median of 7,3,2,6 is 6, and the median of 5,4,8 is 5.
Please help FJ count the number of different contiguous subsequences of his cows that he could potentially submit to the photography contest.
给出一串数字，问中位数大于等于X的连续子串有几个。（这里如果有偶数个数，定义为偏大的那一个而非中间取平均）

输入描述

* Line 1: Two space-separated integers: N and X.
* Lines 2…N+1: Line i+1 contains the single integer HiH_iHi.

输出描述

* Line 1: The number of subsequences of FJ’s cows that have median at least X. Note this may not fit into a 32-bit integer.

示例1

输入

4 6
10
5
6
2

输出

7

说明

FJ’s four cows have heights 10, 5, 6, 2. We want to know how many
contiguous subsequences have median at least 6.
There are 10 possible contiguous subsequences to consider. Of these, only 7
have median at least 6. They are {10}, {6}, {10, 5}, {5, 6}, {6, 2}, {10, 5, 6}, {10, 5, 6, 2}.

解题思路

把大于等于 x 的置为 1 ，小于 x 的置为 −1 ，计算一波前缀和
问题就转化成要找满足 sum[r]−sum[l−1]>=0 的 (l,r) 的对数
枚举这个 l−1 ，即枚举 0 到 n−1 ，每次求 sum[r]>=sum[l−1] 的 r 的个数( 逆序对 )
显然可以用树状数组很方便地维护出来

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+7;

int n,m;
int a[N],b[N],s[N],c[N],tong[N];
ll ans;

int lowbit(int x){
    return (x)&(-x);
}
int change(int x){
    return x+100001;
}
void add(int x,int v){
    while(x<=N){
        c[x]+=v;
        x+=lowbit(x);
    }
}
inline int query(int x){
    int ret=0;
    while(x>0){
        ret+=c[x];
        x-=lowbit(x);
    }
    return ret;
}
signed main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        if(a[i]>=m) b[i]=1;
        else b[i]=-1;
    }
    add(change(0),1);
    tong[change(0)]++;
    for(int i=1;i<=n;i++){
        s[i]=s[i-1]+b[i];
        add(change(s[i]),1);
        tong[change(s[i])]++;
    }
    for(int i=0;i<n;i++){
        int t=n-i-query(change(s[i]))+tong[change(s[i])];
        ans+=t;
        add(change(s[i]),-1);
        tong[change(s[i])]--;
    }
    cout<<ans<<endl;
    return 0;
}