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;
}