LightOJ 1030 Discovering Gold 概率

一道比较简单的概率dp， dp[i]表示取第i个坑里的金子，总共可以得到的金子的期望值， dp[i] = weight[i] + (dp[i + 1] + dp[i + 2] + dp[i + 3] + dp[i + 4] + dp[i + 5] + dp[i + 6]) / 6，注意当最后不足六个的时候要特殊处理。最后dp[1] 就是最终结果。

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>

using namespace std;

int main() {
	int t;
	scanf("%d", &t);

	int a[110];

	double dp[110];

	int CASE = 1;

	while (t--) {
		int n;

		scanf("%d", &n);
		for (int i = 1; i <= n; i++) {
			scanf("%d", a + i);	
		}
		dp[n] = a[n];

		for (int i = n - 1; i > 0; i--) {
			double div = min(6, n - i);	
			double sum = 0;
			for (int j = 1; j <= div; j++) {
				sum += dp[i + j];	
			}
			dp[i] = sum / div + a[i];
			sum += dp[i];
		}
		printf("Case %d: %lf\n", CASE++, dp[1]);

	}
	return 0;
}