51Nod－1682－中位数计数

ACM模版

描述

题解

这里，我们可以分析得到，符合规则的区间有四种形式，分别是：

//      i       (1)
//  j---i       (2)
//      i---j   (3)
//  j'--i--j"   (4)

而这里，第一种不用过多处理，就是1；第2种和第3种类似，所以，我们需要求出来i之前的num的匹配情况，和i之后的num的匹配情况；而第四种要求的是，在第2种和第3种的基础上，进行匹配，匹配成功则符合第4种，这样子，我们把四种情况的结果加在一起也就是答案了。
当然，我们还需要枚举每一个i。

代码

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define MAXSIZE 8008

int num[MAXSIZE];
int arr[MAXSIZE];
int cnt[MAXSIZE];
int sum[MAXSIZE * 2];

//  输入int
void scanfDiy(int *ret)
{
    char c;
    *ret = 0;
    while((c = getchar()) < '0' || c > '9');
    while(c >= '0' && c <= '9')
        *ret = (*ret) * 10 + (c - '0'), c = getchar();
    return ;
}

//  符合规则的区间分为以下几种
//      i       (1)
//  j---i       (2)
//      i---j   (3)
//  j'--i--j"   (4)

int main(int argc, const char * argv[])
{
    int n;

    while (~scanf("%d", &n))
    {
        for (int i = 1; i <= n; i++)
        {
            scanfDiy(num + i);
        }

        for (int i = 1; i <= n; i++)
        {
            int temp = num[i];
            memset(arr, 0, sizeof(arr));
            memset(sum, 0, sizeof(sum));
            int res = 0;

            //  存num[i]左边区间比num[i]大小情况
            for (int j = i - 1; j >= 1; j--)
            {
                arr[j] = arr[j + 1] + ((num[j] > temp) ? -1 : 1);
                sum[arr[j] + n]++;  //  防止越界  相同情况出现的次数
                if (arr[j] == 0)    //  为0时，从j到i区间符合规则
                {
                    res++;          //  (2)
                }
            }
            //  存num[i]右边区间比num[i]大小情况
            for (int j = i + 1; j <= n; j++)
            {
                arr[j] = arr[j - 1] + ((num[j] > temp) ? 1 : -1);
                if (sum[arr[j] + n] > 0)            //  左右匹配
                {
                    res = res + sum[arr[j] + n];    //  (4)
                }
                if (arr[j] == 0)                    //  为0时，从i到j区间符合规则
                {
                    res++;                          //  (3)
                }
            }
            cnt[i] = res + 1;                       //此处+1的情况为(1)
        }

        for (int i = 1; i < n; i++)
        {
            printf("%d ", cnt[i]);
        }
        printf("%d\n", cnt[n]);
    }

    return 0;
}