题目链接: 点这里
题意:一个 1 x N 的格子里,每个格子有若干个金币,一开始在第一个格子里,有一个骰子,每次摇到一个数字x,就从当前格向前移动x格,并取到x格里的金币,求期望取到的金币。
思路: 记忆化搜索。
AC代码:
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <vector>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
#include <cmath>
#define ll long long
using namespace std;
double dp[111];
double a[111];
int n;
double dfs(int x) {
if(x == n){
dp[x] = a[x];
return dp[x];
}
if(dp[x] != 0) return dp[x];
double ans = 0;
if(n-x >= 6) {
for(int i=x+1; i<=(x+6); i++) {
ans += (1.0/6.0)*dfs(i);
}
}else {
int t = n-x;
for(int i=x+1; i<=(x+t); i++) {
ans += (1.0/t)*dfs(i);
}
}
dp[x] = ans+a[x];
return dp[x];
}
int main(){
int t; scanf("%d",&t);
for(int cas = 1; cas <= t; cas++) {
memset(dp,0,sizeof(dp));
scanf("%d",&n);
for(int i=1;i<=n; i++) {
scanf("%lf",&a[i]);
}
dfs(1);
printf("Case %d: %.10lf\n",cas,dp[1]);
}
return 0;
}