D - Average Sleep Time
It's been almost a week since Polycarp couldn't get rid of insomnia. And as you may already know, one week in Berland lasts k days!
When Polycarp went to a doctor with his problem, the doctor asked him about his sleeping schedule (more specifically, the average amount of hours of sleep per week). Luckily, Polycarp kept records of sleep times for the last n days. So now he has a sequence a1, a2, ..., an, where ai is the sleep time on the i-th day.
The number of records is so large that Polycarp is unable to calculate the average value by himself. Thus he is asking you to help him with the calculations. To get the average Polycarp is going to consider k consecutive days as a week. So there will be n - k + 1 weeks to take into consideration. For example, if k = 2, n = 3 and a = [3, 4, 7], then the result is .
You should write a program which will calculate average sleep times of Polycarp over all weeks.
Input
The first line contains two integer numbers n and k (1 ≤ k ≤ n ≤ 2·105).
The second line contains n integer numbers a1, a2, ..., an (1 ≤ ai ≤ 105).
Output
Output average sleeping time over all weeks.
The answer is considered to be correct if its absolute or relative error does not exceed 10 - 6. In particular, it is enough to output real number with at least 6 digits after the decimal point.
Example
Input
3 2
3 4 7
Output
9.0000000000
Input
1 1
10
Output
10.0000000000
Input
8 2
1 2 4 100000 123 456 789 1
Output
28964.2857142857
我的代码：
#include
#include
#include
#include
#include
using namespace std;
#define N 200005
long long int a[N];
int main()
{
long long int n,k,i,j;
double sum = 0,sum1 = 0;
scanf("%lld %lld",&n,&k);
for(int i = 0;i < n; i++)
scanf("%lld",&a[i]);
if(n == k)
{
for(int i = 0;i < n; i++)
{
sum += a[i];
}
printf("%.10lf\n",sum);
}
else
{
j = 0;
for(i = 1;i <= n - k; i++) //n - k表示当有n - k + 1个周时，会有n - k个重复加的数字。
{
j += k - 1;
sum1 += 2 * a[j];
a[j] = 0;
}
for(int i = 0;i < n; i++)
{
if(a[i] != 0)
sum1 += a[i];
}
double ans = sum1 /((n - k + 1) * 1.0);
printf("%.10lf\n",ans);
}
return 0;
}
