Codeforces ~ 1005E（1，2） ~ Median on Segments （思维 or BIT）

最新推荐文章于 2018-11-25 16:31:21 发布

张松超

最新推荐文章于 2018-11-25 16:31:21 发布

阅读量451

点赞数 2

分类专栏：【思维/构造】【BIT/ST表/线段树】

本文链接：https://blog.csdn.net/ZscDst/article/details/80991896

版权

【思维/构造】同时被 2 个专栏收录

73 篇文章 0 订阅

订阅专栏

【BIT/ST表/线段树】

7 篇文章 0 订阅

订阅专栏

题意

n个数字，问m是多少个区间的中位数？偶数个数字的区间中位数认为是中间靠左的那个数字，比如(1,3,7,8)中位数为3。

E1：这n个数字是1~n的一个排列

E2：n个数字，数字大小不超过2e5

题解

做法①(可过E1，过不了E2)：

思路请看：51Nod ~ 1682 ~ 中位数计数（思维）

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 2e5+5;
const int FIX = 2e5;
typedef long long LL;
int n, m, a[MAXN], pos, b[MAXN*2];
int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &a[i]);
        if (a[i] == m) pos = i;
    }
    int cnt = 0;
    LL ans = 0;
    for (int i = pos; i >= 0; i--)
    {
        if (a[i] > m) cnt++;
        if (a[i] < m) cnt--;
        b[FIX+cnt]++;
    }
    cnt = 0;
    for (int i = pos; i < n; i++)
    {
        if (a[i] > m) cnt++;
        if (a[i] < m) cnt--;
        ans += b[FIX-cnt];
        ans += b[FIX-cnt+1];
    }
    printf("%lld\n", ans);
    return 0;
}
/*
5 4
2 4 5 3 1
*/

做法②（E1，2均可过)：

中位数小于等于m的区间 - 小于等于m-1的区间=中位数等于m的区间，现在我们需要搞一个计算中位数小于等于x的区间个数的函数。首先如果中位数小于等于x，那么<=x的数字一定大于>x的数字，我们记<=x的为+1，>x的数字记为-1，我们用变量s记录这个值，如果对于[L,R]区间（L<R），L位置时的s小于R位置时的s，证明[L,R]区间是一个符合要求的区间。

枚举右端点，维护s变量，对于右端点 R，怎么得到s_L <= s_R 的 L 的个数呢？

O（n）做法：

数组cnt[i]用于统计 s = i 的点的个数（由于s有负值所以我们以n作为0点即可），add维护s_L <= s_i 的 L 的个数

如果当前元素a[i]<=x，s要++，那么对于 i 位置符合要求的点比 i-1 多了cnt[++s]个

如果当前元素a[i]>x，s要--，那么对于 i 位置符合要求的点比 i-1 少了cnt[s--]个

统计答案，每次ans+=add即可，最终ans就是中位数大于等于x的区间数。

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 2e5+5;
typedef long long LL;
int n, m, a[MAXN], cnt[MAXN*2];
LL slove(int x)
{
    memset(cnt, 0, sizeof(cnt));
    int s = n; cnt[n] = 1;
    LL add = 0, ans = 0;
    for (int i = 0; i < n; i++)
    {
        if (a[i] <= x) add += cnt[++s];//add += cnt[++s]同s++; add += cnt[s];
        else add -= cnt[s--];//add -= cnt[s--]同add -= cnt[s]; s--;
        cnt[s]++;
        ans += add;
    }
    return ans;
}
int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 0; i < n; i++) scanf("%d", &a[i]);
    printf("%lld\n", slove(m) - slove(m-1));
    return 0;
}
/*
5 4
1 4 5 60 4
*/

O（n*logn）（BIT做法）：

数组cnt[i]用于统计 s = i 的个数（由于s有负值所以我们以n作为0点即可）

BIT维护cnt[s]这个数组，枚举右端点，每次对于当前右端点，查询有cnt中有多少个小于等于s的值。

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 2e5+5;
typedef long long LL;
int n, m, a[MAXN];
struct BIT
{
    int n, c[MAXN*2];
    void init (int n)
    {
        this->n = n;
        memset(c, 0, sizeof(c));
    }

    void add(int p, int x)
    {
        for (int i = p; i <= n; i += i&-i)
            c[i] += x;
    }

    int sum(int p)
    {
        int ans = 0;
        for (int i = p; i >= 1; i -= i&-i)
            ans += c[i];
        return ans;
    }
}bit;
LL slove(int x)
{
    bit.init(2*n+1);
    bit.add(n+1, 1);
    int s = n+1;
    LL ans = 0;
    for (int i = 1; i <= n; i++)
    {
        if (a[i] <= x) s++;
        else s--;
        bit.add(s, 1);
        ans += bit.sum(s);
    }
    return ans;
}
int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
    printf("%lld\n", slove(m) - slove(m-1));
    return 0;
}
/*
5 4
1 4 5 60 4
*/