题目链接:题目传送门
概率dp刷的比较少,这个题看了好几天看不懂,真的是捉急啊
题目描述
给你一个长 n 的序列,m 次查询
每次查询给一个 x,然后:
从序列的最左端 1 开始,每次随机的选择一个右端点 r,如果两个端点间的区间和不超过 x ,就进行一次分割,然后把左端点变成 r + 1, 否则一直随机下去。
问这样分割出来的期望段数
输入描述:
第一行两个数 n,m
之后一行 n 个数表示这个序列
之后m行每行一个数 x,表示求每段的和不大于 x 的情况下切割的期望段数
输出描述:
m 行,每行一个保留到小数点后2位的数表示答案
如果无论如何都没有合法的切割方案,输出"YNOI is good OI!",不含引号
示例1
输入
5 6
1 2 3 4 5
4
5
6
7
8
9
输出
YNOI is good OI!
4.25
3.83
3.58
3.58
3.11
说明
对于30%的数据,n , m <= 10
对于60%的数据,n <= 1000 , m <= 10
对于80%的数据,n <= 100000 , m <= 100
对于100%的数据,n <= 100000 , m <= 500 , 0 <= a[i] <= 1000 , x <= 1000000
PS:数据直接随的,不知道有没有乱搞方法~
思路:
求出 i 位置的断开的概率 pi,那么答案就是 ans=∑ni=1pi
假设知道了 pi , 同时也可以知道这个位置之后最多连续有多少个元素和小于 x 的的元素个数 inum,那么这位对 [i+1,i+2,…,i+inum] 贡献 p iinum
那么对于每一位
pi=∑k<i≤k+knumpkknum
尺取 O(n)
代码:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <set>
#include <map>
#include <stack>
#include <vector>
#include <queue>
#define ri(n) scanf("%d",&n)
#define oi(n) printf("%d\n",n)
#define rl(n) scanf("%lld",&n)
#define ol(n) printf("%lld\n",n)
#define rep(i,l,r) for(i=l;i<=r;i++)
#define rep1(i,l,r) for(i=l;i<r;i++)
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int epg=10-8;
const int maxn=1e5+10;
int a[maxn];
double o[maxn];
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)==2)
{
int ma=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
ma=max(ma,a[i]);
}
while(m--)
{
int x;
scanf("%d",&x);
if(x<ma)
{
printf("YNOI is good OI!\n");
continue;
}
else
{
for(int i=0;i<=n;i++)
o[i]=0;
double s=0,p=0,pp=1.0;
double ans=0;
int low=1,top=0;
while(low<=n)
{
if(s<=x&&top<=n)
s+=a[++top];
else
{
double t=pp/(top-low);
o[top]-=t;
s-=a[low];
p+=o[low++]+t;
ans+=pp=p;
}
}
printf("%.2f\n",ans);
}
}
}
return 0;
}