HDU 4908 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.csdn.net/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;
}