描述
题解
这道题坑挺多,挺烦人的,精度问题,需要用 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;
}