BZOJ 1303 [CQOI2009]中位数图（数学）

最新推荐文章于 2018-12-23 15:50:00 发布

pengwill97

最新推荐文章于 2018-12-23 15:50:00 发布

阅读量157

点赞数

本文链接：https://blog.csdn.net/pengwill97/article/details/80403535

版权

BZOJ 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

题意分析

比 $k$ 小权值为-1，比 $k$ 大权值为1。
以 k <script type="math/tex" id="MathJax-Element-6">k</script>为中心，向左向右算个前缀和，统计一下个数，统计时别忘了shift。
然后从左查询，右区间查询为0的即可。

代码总览

#include <bits/stdc++.h>
#define rep(i,a,b) for (int i = a; i<=b; ++i)
using namespace std;
const int nmax = 1e5 + 100;
const int INF = 0x3f3f3f3f;
const int shift = 1e5;
typedef long long ll;
typedef double db;
int a[nmax], tag[nmax], Lcnt[nmax << 1], Rcnt[nmax << 1], Lprefix[nmax], Rprefix[nmax];
int n, k, pos;
int main() {
    scanf("%d %d", &n, &k);
    rep(i, 1, n) scanf("%d", &a[i]);
    rep(i, 1, n) {
        if (a[i] < k) tag[i] = -1;
        else if (a[i] == k) tag[i] = 0, pos = i;
        else tag[i] = 1;
    }
    int ans = 1;
    int sum = 0; Lcnt[0 + shift] = Rcnt[0 + shift] = 1;
    for (int i = pos - 1; i >= 1; --i) {
        sum += tag[i]; Lprefix[i] = sum;
        Lcnt[sum + shift] ++;
    }
    sum = 0;
    rep(i, pos + 1, n) {
        sum += tag[i]; Rprefix[i] = sum;
        Rcnt[sum + shift]++;
    }
    if (pos == 1) {
        rep(i, 2, n) ans += Lcnt[-Rprefix[i] + shift];
    } else if (pos == n) {
        rep(i, 1, n - 1) ans += Rcnt[-Lprefix[i] + shift];
    } else {
        rep(i, 1, pos - 1) ans += Rcnt[-Lprefix[i] + shift];
        rep(i, pos + 1, n) if (Rprefix[i] == 0) ans++;
    }
    printf("%d\n", ans);
    return 0;
}