HDU 4908 BestCoder Sequence

最新推荐文章于 2020-07-12 20:18:12 发布

原创最新推荐文章于 2020-07-12 20:18:12 发布 · 1.9k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#ACM

=====算法相关===== 同时被 2 个专栏收录

319 篇文章

订阅专栏

+基础算法

143 篇文章

订阅专栏

本文介绍了解决HDU4908 BestCoder Sequence问题的方法，该问题要求找出包含特定中位数的所有奇数长度子序列的数量。通过记录序列中目标值左右两侧大于或小于它的元素数量差异，实现了一种高效的解决方案。

HDU 4908 BestCoder Sequence

ACM

题目地址：HDU 2276 BestCoder Sequence

题意：
给一个序列，里面是1～N的排列，给出M，问以M为中位数的奇数长度的序列个数。

分析：
就是记录M左右两边区间比M大的和比M小的差值的个数，再相乘。
可以先遍历左边记录差值个数，右边遍历直接加上左边的一样的差值。
比赛时已经想到做法了却老是没敲出来，太弱了。

代码：

/*
*  Author:      illuz <iilluzen[at]gmail.com>
*  Blog:        http://blog.youkuaiyun.com/hcbbt
*  File:        4908.cpp
*  Create Date: 2014-08-04 09:14:45
*  Descripton:   
*/

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define repf(i,a,b) for(int i=(a);i<=(b);i++)

typedef long long ll;

const int N = 4e4+10;

int a[N], rec[N<<1];
int n, m, pos, cnt;
ll ans;

int main() {
    while (~scanf("%d%d", &n, &m)) {
        memset(rec, 0, sizeof(rec));
        repf (i, 0, n - 1) {
            scanf("%d", &a[i]);
            if (a[i] == m) {
                pos = i;
            }
        }
        cnt = 0;
        for (int i = pos; i >= 0; i--) {
            if (a[i] > m) cnt++;
            if (a[i] < m) cnt--;
            rec[N + cnt]++;
        }
        cnt = 0;
        ans = 0;
        for (int i = pos; i < n; i++) {
            if (a[i] > m) cnt++;
            if (a[i] < m) cnt--;
            ans += rec[N - cnt];
        }
        cout << ans << endl;
    }
    return 0;
}