P1627 中位数

P1627 中位数

题目描述

给出1~n的一个排列，统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后，位于中间的数。

输入输出格式

输入格式：

第一行为两个正整数n和b，第二行为1~n的排列。

【数据规模】

对于30%的数据中，满足n≤100；

对于60%的数据中，满足n≤1000；

对于100%的数据中，满足n≤100000,1≤b≤n。

输出格式：

输出一个整数，即中位数为b的连续子序列个数。

输入输出样例

输入样例#1：

7 4
5 7 2 4 3 1 6 

输出样例#1：

4

分析

　　首先大于b的设为1，小于b的设为-1，p点为b值的坐标。处理L数组和R数组，L[i]表示p左边有多少个点到p的和为i，R[i]表示p右边有多少个点到p的和为i，那么ans=Σ L[i]*R[0-i]。为了没有负数坐标，所以代码中统一加n。

code

#include<cstdio>

int a[100100];
int l[200100],r[200100],sum[100100];
int read()
{
    int x = 0, f = 1; char ch = getchar();
    for (; ch<'0'||ch>'9'; ch = getchar())
        if (ch=='-') f = -1;
    for (; ch>='0'&&ch<='9'; ch = getchar())
        x = x*10+ch-'0';
    return x*f;
}
int main()
{
    int n = read(), d = read(), p,ans = 0;
    for (int x,i=1; i<=n; ++i) {
        x = read();
        if (d==x) p = i,a[i] = 0;
        else if (x<d) a[i] = -1;
        else a[i] = 1;
    }
    l[n] = r[n] = 1;
    for (int i=p-1; i>=1; --i) {
        sum[i] = sum[i+1]+a[i];
        l[sum[i]+n]++;
    }
    for (int i=p+1; i<=n; ++i) {
        sum[i] = sum[i-1]+a[i];
        r[sum[i]+n]++;
    }
    for (int i=0; i<=2*n-1; ++i) ans += l[i]*r[2*n-i];
    printf("%d",ans);    
    return 0;
}

乱搞70

#include<cstdio>
#include<algorithm>


using namespace std;

int a[10000100];

int read()
{
    int x = 0, f = 1; char ch = getchar();
    for (; ch<'0'||ch>'9'; ch = getchar())
        if (ch=='-') f = -1;
    for (; ch>='0'&&ch<='9'; ch = getchar())
        x = x*10+ch-'0';
    return x*f;
}

int main()
{
    int n = read(), d = read(), p = -1, xcnt = 0, dcnt = 0,ans = 1;
    
    for (int i=1; i<=n; ++i) {
        a[i] = read();
        if (d==a[i]) p = i;
        if (p==-1) {
            if (a[i]<d) xcnt++;
            else dcnt++;
        }
    }
    int L = p,da = 0,xo = 0;
    for (int i=1; ; i+=2) {
        if (a[--L]>d) da++;else xo++;
        if (a[--L]>d) da++;else xo++;
        if (da==xo)    ans++;
        if (L==1||L==2) break;
    }
    
    for (int i=1; i<=p; ++i) {
        if (i!=1) {
            if (a[i-1]>d) dcnt--;
            else xcnt--;
        }
        da = dcnt,xo = xcnt;
        for (int j=p+1; j<=n; ++j) {
            if (a[j]>d) da++;else xo++;
            if (da==xo) ans++;
        }    
    }
    
    printf("%d",ans);
        
    return 0;
}