HDU 4405(概率DP)

好久没写博客了，被队友嫌弃了，正好现在在补题，顺便多写点。

大致题意：一个飞行棋游戏，从0到N，你有一个骰子，每次掷骰子，等概率前进1-6步，但是飞行棋中有m条捷径，（a，b），即如果走到a即可以直接前进到b，这个前进是不间断的，比如有两条捷径（a，b），（b，c），那么如果你走到a，就可以直接前进到c，最后求掷骰子的期望。

这道题好像是可以直接拿概率公式推导出来的，但是弱菜并不会，说一下我自己的思路。

先考虑一种简单的算法，即开一个二维dp[i][j]，表示当前走到第i个点已经投了j次的概率是多少，这个状态比较好转移，dp[i+l][j] += dp[i][j] / 6.0 (1<= l <= 6)，这样n*n扫一遍过去就可以了，最后的结果就是第N个位置的期望。但是这个题的数据量是100000，这个算法的空间和时间复杂度明显不行，于是我们来想办法把这个dp优化成一维的。

下面表格是模拟的二维矩阵的一些运算：

投掷次数	行进到第i个位置的概率	行进到第j个位置的概率
0	P0	Q0
1	P1	Q1 + P0 / 6
2	P2	Q2 + P1 / 6
3	P3	Q3 + P2 / 6
4	P4	Q4 + P3 / 6
……	总概率P，期望Ei	总概率Q，期望Ej

根据期望公式，Ei=0*P0+1*P1+2*P2……，E(j+)=1*P0/6+2*P1/6+3*P2/6……(Ej增加的部分)即

6*E(j+) - Ei = P0 + P1 + P2 +  ……= P 

E(j+) = (Ei + P) / 6

Q += P

这样，我们就成功的加二维dp中一些重复运算的地方成功优化掉了，复杂度为O（n），详细代码如下：

#include <cstdio>
#include <iostream>
#include <string.h>
#include <cmath>
#include <algorithm>
#include <map>
using namespace std;
const int maxn = 100010;
map<int, int> haha;
double dp[maxn], p[maxn];
int n, m;
int main() {
    while (scanf("%d %d", &n, &m) == 2 && (n + m)) {
        memset(dp, 0, sizeof dp);
        memset(p, 0, sizeof p);
        p[0] = 1.0;
        haha.clear();
        for (int i = 0; i < m; ++i) {
            int a, b;
            scanf("%d %d", &a, &b);
            haha[a] = b;
        }
        for (int i = 0; i < n; ++i) {
            if (haha.count(i)) {
                dp[haha[i]] += dp[i];
                p[haha[i]] += p[i];
                continue;
            }
            for (int l = 1; l <= 6; ++l) {
                int nb;
                if (i + l < n) nb = i + l;
                else nb = n;
                dp[nb] += (dp[i] + p[i]) / 6.0;
                p[nb] += p[i] / 6.0;
            }
        }
        printf("%.4f\n", dp[n]);
    }
    return 0;
}