码迷,mamicode.com
首页 > 其他好文 > 详细

hdu 4908

时间:2015-07-29 19:26:51      阅读:170      评论:0      收藏:0      [点我收藏+]

标签:acm

题意

给你1-n的数,求以m为中位数的子序列有多少个

思路

l【i】表示m左边综合起来比m大的数个数为i
r【i】表示m左边综合起来比m小的数个数为i
那么在右面就可以找到综合起来比m小的个数为i的加上r【i】
比他大的加上l【i】

代码

/* **********************************************
Auther: 请叫我acm渣渣
Created Time: 2015-7-29 16:58:53
File Name   : hdu4908.cpp
*********************************************** */
#include <iostream>
#include <fstream>
#include <cstring>
#include <climits>
#include <deque>
#include <cmath>
#include <queue>
#include <stack>
#include <list>
#include <map>
#include <set>
#include <utility>
#include <sstream>
#include <complex>
#include <string>
#include <vector>
#include <cstdlib>
#include <cstdio>
#include <ctime>
#include <bitset>
#include <functional>
#include <algorithm>
using namespace std;
#define lson L,m,rt<<1
#define rson m+1,R,rt<<1|1
#define ll long long
#define N 111111

int n,m;
int sum ;
int a[N];
int b[N];
int l[N],r[N];
int main(){
    while(scanf("%d%d",&n,&m)==2){
        int mid = 0;
        memset(l,0,sizeof(l));
        memset(r,0,sizeof(r));
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            if(a[i]==m) mid = i;
        }
        int cur =  0;
        for(int i=mid;i>0;i--){
            if(a[i]>m) cur++;
            if(a[i]<m) cur--;
            if(cur>0) l[cur]++;
            else r[-cur]++;
        }
        int ans= 0 ;
        cur = 0;
        for(int i=mid;i<=n;i++){
            if(a[i]>m) cur++;
            if(a[i]<m) cur--;
            if(cur>=0) ans+=r[cur];
            else ans+=l[-cur];
        }

        printf("%d\n",ans);
    }
} 

版权声明:都是兄弟,请随意转载,请注明兄弟是谁

hdu 4908

标签:acm

原文地址:http://blog.csdn.net/u013076044/article/details/47130811

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!