筛子游戏 期望dp 大力推柿子

题目链接

首先三个骰子的和是i的概率可以预处理出来

三层for循环暴力枚举即可，每一次的概率都是1/a[1]*a[2]*a[3]

我们设dp[i]为i到n需要的期望步数

p[i]为骰到i的概率

那么对于i，很显然，dp[i]=对于所有可能的k ：dp[i+k]*p[k]加上dp[0]乘以p[0]+1

我们整理成dp[i]=a[i]*dp[0]+b[i]的形式

最后结果就是b[0]*a[0]/(1-a[0]);

在这里插入代码片#include <bits/stdc++.h>
using namespace std;
const int N = 1e6+5;
#define int long long
const int INF = 4e18+5;
#define F(i,l,r) for(int i=l;i<=r;i++)
#define R(i,l,r) for(int i=r;i>=l;i--)
#define vv vector
#define fi first
#define se second
#define pii pair<int,int>
typedef long long ll;
const int mod = 998244353; 
//const int mod = 1e9+7; 
const int M = 1e7+5;
int a[N],b[N],c,d,vis[N],pri[N],ans1[N],ans2[N],p[N]; 
int dx[4]={1,0,-1,0};
int dy[4]={0,1,0,-1};
int n,m,k,ans,x,y,n0,n2,m0,m2,op1,op2;
string s;
map<pii,int>mp;
map<int,int>mmap;
double aa[N],bb[N],dp[N];//到达点i的概率 
//char mp[1005][1005];
//vector<int>v1[N],v2[N];
int lowbit(int x) {
	return x&-x;
}
char xx[N];
bool temp;
int kuai(int x,int y){
	int sum=1;
	x%=mod;
	while(y){
		if(y%2){
			y--;
			sum*=x;
			sum%=mod;
		  }
	  x*=x;
	  y/=2;
	  x%=mod;
	}
	return sum;
}
struct node{
	int shu;
	int id;
}jie[N];
bool cmp(node x,node y){
	return x.shu<y.shu;
}
void solve()
{
	cin>>n>>a[1]>>a[2]>>a[3]>>b[1]>>b[2]>>b[3];
	F(i,1,a[1]) F(j,1,a[2]) F(k,1,a[3]) p[i+j+k]++;
	int zhong=a[1]*a[2]*a[3];//去除之外总共的以及全部总共的 
	p[b[1]+b[2]+b[3]]--;
	R(i,0,n){//i这个点 
		for(int j=3;j+i<=n&&j<=a[1]+a[2]+a[3];j++){//i这个点可以从哪些点跳过来 
			if(p[j]){
				aa[i]+=1.0*aa[i+j]*p[j]/zhong;
				//aa[i]+=1.0/zhong;
				bb[i]+=1.0*bb[i+j]*p[j]/zhong;
			}
		}
		aa[i]+=1.0/zhong;
		bb[i]+=1.0;
	}
	double xx=bb[0]/(1-aa[0]);
	printf("%.10lf",xx);
}
signed main() 
{
	ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0); // cin.tie(nullptr);
    int T = 1;                                                                                                                                                                                                                                                                    
    //cin >> T;
    for (int i = 1; i <= T; i++)
    {
        solve();
    }
    return 0;
}