1. 程式人生 > >BZOJ1303: [CQOI2009]中位數圖

BZOJ1303: [CQOI2009]中位數圖

post 中位數 inpu sample str gpo turn rip markdown

Description

給出1~n的一個排列,統計該排列有多少個長度為奇數的連續子序列的中位數是b。中位數是指把所有元素從小到大排列後,位於中間的數。
Input

第一行為兩個正整數n和b ,第二行為1~n 的排列。
Output

輸出一個整數,即中位數為b的連續子序列個數。
Sample Input

7 4

5 7 2 4 3 1 6
Sample Output

4
HINT

第三個樣例解釋:{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}
N<=100000

題目傳送門

一開始死活不會搞
這是一個全排列嘛,所以b只有一個,還要求是奇數。。。
那這個序列肯定包含b啊。。。
那麽把b設為0,>b設為-1,

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
int a[110000],l[210000],r[210000];
int main() 
{
    int n,b,pos;scanf("%d%d",&n,&b);
    for(int i=1;i<=n;i++) {
        int x;scanf("%d",&x);
        if(x<b)a[i]=1;
        if
(x>b)a[i]=-1; if(x==b)pos=i; } int sum=0;l[n]=r[n]=1; for(int i=pos-1;i>=1;i--) sum+=a[i],l[sum+n]++; sum=0; for(int i=pos+1;i<=n;i++) sum+=a[i],r[sum+n]++; int ans=0; for(int i=-n;i<=n;i++) ans+=l[n-i]*r[i+n]; printf("%d
\n"
,ans); return 0; }

by_lmy

BZOJ1303: [CQOI2009]中位數圖