POJ 2018
又一水,设dp[i]为以i结尾的有最大平均值的起始位置。
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #include <climits> #include <string.h> #include <queue> #include <cmath> #include <vector> using namespace std; int cow[100010]; int dp[100010]; int n,f; int main(){ while(scanf("%d%d",&n,&f)!=EOF){ cow[0]=0; for(int i=1;i<=n;i++){ scanf("%d",&cow[i]); cow[i]+=cow[i-1]; } dp[f]=1; int pos=f; double ans1,ans2,ans3; for(int i=f+1;i<=n;i++){ ans1=(cow[i]-cow[dp[i-1]-1])*1.0/(i-dp[i-1]+1); ans2=(cow[i]-cow[i-f])*1.0/(f); if(ans1>ans2){ dp[i]=dp[i-1]; } else dp[i]=i-f+1; ans3=(cow[pos]-cow[dp[pos]-1])*1.0/(pos-dp[pos]+1); ans1=(cow[i]-cow[dp[i]-1])*1.0/(i-dp[i]+1); if(ans1>ans3) pos=i; } printf("%d\n",int((cow[pos]-cow[dp[pos]-1])*1000.0/(pos-dp[pos]+1))); } return 0; }