蓝桥杯 算法提高 概率计算 （概率DP）

  算法提高 概率计算  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

　　生成n个∈[a,b]的随机整数，输出它们的和为x的概率。

输入格式

　　一行输入四个整数依次为n，a，b，x，用空格分隔。

输出格式

　　输出一行包含一个小数位和为x的概率，小数点后保留四位小数

样例输入

2 1 3 4

样例输出

0.3333

数据规模和约定

　　对于50%的数据，n≤5.
　　对于100%的数据，n≤100,b≤100.

ac代码

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define LL long long
using namespace std;
double dp[110][5500];
double DP(LL m,LL a,LL b,LL n,LL sum)
{
    int i,j,t;
    memset(dp,0,sizeof(dp));
    dp[0][0]=1;
    for(i=1;i<=m;i++)
    {
        for(j=0;j<=n;j++)
        {
            for(t=a;t<=b&&t<=j;t++)
            {
                dp[i][j]+=dp[i-1][j-t]/sum;
            }
        }
    }
    return dp[m][n];
}
int main()
{
    LL n,a,b,x;
    while(scanf("%lld%lld%lld%lld",&n,&a,&b,&x)!=EOF)
    {
        LL sum=b-a+1;
        double temp=DP(n,a,b,x,sum);
        printf("%.4lf\n",temp);
    }
}