51Nod-1785-数据流中的算法

ACM模版

描述

题解

这道题坑挺多，挺烦人的，精度问题，需要用 double，并且，虽然保留两位小数，但是均值必须是整型，不过输出需要保留两位小数，实际上也就是最后多加一个 .00 而已。直接暴力解就可以 AC，不过需要进行一些优化，也不能太暴力了，需要一些技巧，具体看代码吧！对了，如果暴力搞，还需要输入外挂，否则会 TLE。

当然，也有其他办法，求中位数的部分可以使用线段树搞搞事情。

一开始我有两组数据莫名其妙的 WA，下载下来数据百般尝试，本地测试结果都是对的，搞得我十分纳闷儿，最后猛然想起来，以前也遇见这种情况，明明只让保留两位或者三位的小数，float 却不行，于是改成了 double 一试，果然 AC 了~~~

另外，<stdio.h>要比<cstdio>快一些，有的人因为这个而 TLE 了。

代码

#include <stdio.h>

const int MAXN = 1e6 + 10;
const int MAXK = 105;

int n, k, cnt = 0, sum = 0, temp = 0;
int f[MAXN];
int f_[MAXK];

inline void read(int &ret)
{
    char c;
    ret = 0;
    while ((c = getchar()) < '0' || c > '9');
    while (c >= '0' && c <= '9')
    {
        ret = ret * 10 + (c - '0'), c = getchar();
    }
}

//  方差
double variance()
{
    double ave = sum * 1.0 / temp;

    double var = 0;
    for (int i = cnt; i > cnt - temp; i--)
    {
        var += (f[i] - ave) * (f[i] - ave);
    }

    return var / temp;
}

//  中位数
double midNum()
{
    int n = 0, a = -1, b = -1;
    int t = temp / 2, t_ = t + 1;

    for (int i = 0; i <= 100; i++)
    {
        n += f_[i];
        if (n >= t && a == -1)
        {
            a = i;
        }
        if (n >= t_ && b == -1)
        {
            b = i;
            break;
        }
    }

    if (temp & 1)
    {
        return b;
    }
    else
    {
        return (a + b) * 1.0 / 2;
    }
}

int main(int argc, const char * argv[])
{
//    freopen("/Users/zyj/Desktop/input.txt", "r", stdin);

    read(n);
    read(k);

    int u;
    for (int i = 1; i <= n; i++)
    {
        read(u);
        if (u == 1)
        {
            read(f[++cnt]);
            if (cnt > k)
            {
                f_[f[cnt - k]]--;
                sum -= f[cnt - k];
            }
            f_[f[cnt]]++;
            sum += f[cnt];
            temp = cnt > k ? k : cnt;
        }
        else if (u == 2)
        {
            double num = sum / temp;
            printf("%.2lf\n", num);
        }
        else if (u == 3)
        {
            double var = variance();
            printf("%.2lf\n", var);
        }
        else
        {
            double mid = midNum();
            printf("%.2lf\n", mid);
        }
    }

    return 0;
}