HYSBZ

来源:互联网 发布:php二维数组去掉重复值 编辑:程序博客网 时间:2024/04/28 19:00

题意:

给定 1-n 的排列,找到以 b 为中位数的区间(奇数长度)个数;

思路:

这个区间要包含 b,可能是b及其左边的一段区间,或者b及其右边的一段区间,或者左右两边的区间组合;

开始没有看见 1-n 的排列组合,以为是任意的数,那样不容易搞,后来发现1-n这个排列以后就很简单了,

这是这个区间满足b是中位数这个条件的话,那大于他的数的个数等于小于他的数到个数;

左右两边分别计数, On 的复杂度;


#include<iostream>#include<algorithm>#include<cstdio>#include<cstdlib>#include<cstring>#include<string>#include<cmath>#include<set>#include<queue>#include<stack>#include<map>#define PI acos(-1.0)#define in freopen("in.txt", "r", stdin)#define out freopen("out.txt", "w", stdout)using namespace std;typedef long long ll;typedef unsigned long long ull;const int maxn = 1e5 + 7, maxd = 670000 + 7, mod = 1e9 + 7;const int INF = 0x7f7f7f7f;int n, b, id;int a[maxn];int l_[maxn], r_[maxn];map<int, int> mp;int main() {    ios::sync_with_stdio(0);    cin >> n >> b;    for(int i = 1; i <= n; ++i) {        cin >> a[i];        if(a[i] == b) id = i;    }    int cnt = 0;    for(int i = id-1; i > 0; --i) {        if(a[i] > b) l_[i] = (++cnt);        if(a[i] < b) l_[i] = (--cnt);    }    cnt = 0;    for(int i = id+1; i <= n; ++i) {        if(a[i] > b) r_[i] = (++cnt);        if(a[i] < b) r_[i] = (--cnt);    }    int ans = 1;    for(int i = id-1; i > 0; --i) {        if(l_[i] == 0) ans ++;        mp[l_[i]]++;    }    for(int i = id+1; i <= n; ++i) {        if(r_[i] == 0) ans++;        if(mp.count(-r_[i])) ans += mp[-r_[i]];    }    cout << ans << endl;    return 0;}


原创粉丝点击