Wannafly挑战赛3 珂学送分(期望dp)

科学送分

定义$dp[i]$为$[i,n]分段的期望$

那么$dp[n]=1$

$对于确定的i,设我们找到了最大的j满足\sum _{k=i}^j{a}_k<x$

$说明i可以分裂到[i,j]上的任意一个点去,那么有j-i+1种可能$

$在[i,j]上任取一点k表示[i,k]分成一段,贡献期望是\frac{1}{j-i+1}\ast dp[k+1]$

$又发现对于i来说,如果i递减,j也递减,所以可以维护后缀和来解决$

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
int a[maxn],pre[maxn],n,m,maxx;
double dp[maxn],su[maxn]; 
int main()
{
	cin >> n >> m;
	for(int i=1;i<=n;i++)
	{
		cin >> a[i];
		pre[i]=pre[i-1]+a[i];
		maxx=max( maxx,a[i] );
	}
	while( m-- )
	{
		int x; cin >> x;
		if( maxx>x )
		{
			printf("YNOI is good OI!\n");
			continue;
		}
		su[n]=1,su[n+1]=0;
		dp[n]=1;
		int j=n;
		for(int i=n-1;i;i-- )
		{
			while( pre[j]-pre[i-1]>x )	j--;
			//i位置可以转移到[i,j] 
			dp[i]=1.0/( j-i+1 )*( su[i+1]-su[j+2] )+1.0;
			su[i]=su[i+1]+dp[i];
		}
		printf("%.2f\n",dp[1]);
	}
}