HDU dice 期望DP

题意：
有一个m面的骰子，且质量均匀（掷骰子出现每一面的概率相同），问： 掷出n次结果都相同的期望投掷次数和n次结果都不同的期望投掷次数。

分析：
第一种：
设dp【i】来表示已经有连续i个相同面时，要到达目标状态的期望次数。
①.下一次投掷和前i次都相同的面，dp【i】= (dp【i】+ 1) * (1.0/m)
②.下一次投掷和前i次都不一样的面, 则需要从第一次开始算 dp【i】= (dp【1】+ 1) * (m - 1.0) / m
综合：dp【i】= (dp【i+1】) * 1.0 / m + dp【1】* (m - 1) / m + 1
–> m * dp【i】= m + (m - 1) * dp【1】+ dp【i+1】
–> m * dp【i+1】= m + (m - 1) * dp【1】+ dp【i+2】
相减得：m * (dp【i+1】 - dp【i】) = dp【i+2】 - dp【i+1】
dp【0】- dp【1】= 1
dp【1】- dp【2】= m
…
dp【n-1】- dp【n】= m^n-1
裂项相消 等比数列求和：res = (mⁿ - 1) / (1 - m)

第二种:
设dp【i】来表示已经有连续i个不相同面时，要到达目标状态的期望次数。
①.下一次投掷与前i次都不同，转移到dp【i+1】: dp【i】= (dp【i+1】+1) * (m-i) / m
②.下一次投掷与前i次有相同, 可能转移到dp【j】(j = 1, 2, … i) : dp【i】= (dp【1】+ dp【2】+…+dp【i】
与第一种同样操作后即可

代码如下

#include<iostream>
#include<cmath>

using namespace std;
#define ll long long
#define sqr(x) (x)*(x)
const int maxn = 1e6+10;
double fun(int m, int n){
	double res = 1;
	double tmp = 1;
	for(int i=1; i<n; i++){
		tmp = 1.0 * m / (m - i) * tmp;
		res += tmp;
	}
	return res;
}
int main(){
	int t;
	while(~scanf("%d", &t)){
		while(t--){
			int op, n, m;
			scanf("%d %d %d", &op, &m, &n);
			if(op == 0){
				double res = (pow(m, n) - 1.0) / (m - 1.0);
				printf("%.9lf\n", res);
			}else{
				printf("%.9lf\n", fun(m, n));
			}
		}
	}
	return 0;
}