HDU 4035 Maze 树形概率DP 2011年成都网络赛E题

题目大意:
lxhgww 现在在一个树形图的点1上(此树形图共n个点,编号从1到n) 现在在第i点有 Ki%的概率被杀死回到点1, 有Ei%的概率逃出迷宫, 剩下的1 - Ki% - Ei%就是什么都没发生了,需要转到下一位置,其中K0 = E0 = 0, 现在给出树形图,定点数n <= 10000求lxhgww走出迷宫需要走多少次的期望


大致思路:
由于n <= 10000直接解线性方程组的复杂度不能接受,需要递推,递推公式及过程见代码


代码如下:

Result  :  Accepted     Memory  :  1588 KB     Time  :  203 ms

/*
 * Author: Gatevin
 * Created Time:  2014/12/2 17:57:35
 * File Name: Asuna.cpp
 */
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
const double eps(1e-10);//这里刚开始开eps = 1e-8 WA了...
typedef long long lint;

/*
 * 很好的一道概率题, 首先不难发现如果用dp[i]表示从节点i到达目标状态需要的步数,有方程:
 * dp[i] = ∑(dp[j] + 1)/G[i].size()*(1 - K[i] - E[i]) + K[i]*(dp[1] + 0) + E[i]*0;
 *       = ∑(dp[j] + 1)/G[i].size()*(1 - K[i] - E[i]) + K[i]*dp[1]
 * 其中G[i]为与点i相邻的点组成的vector, K[i]为在i节点被杀死的概率, E[i]为出去的概率,j为与i相邻的点
 * 这样对于dp[1~n]一共有n个方程,可见一定是可解的,但是由于n的范围太大,需要递推求解
 * 这里由于dp[1]为起点,将其视作整棵树的根节点,那么原来的方程需要写为
 * dp[i] = ∑dp[son[i]]/G[i].size()*(1 - K[i] - E[i]) + dp[father[i]]/G[i].size()*(1 - K[i] - E[i]) + K[i]*dp[1] + (1 - K[i] - E[i])
 * 其中son[i]是i的子节点,father[i]是i的父节点,那么对于叶子节点
 * dp[i] = dp[father[i]]*(1 - K[i] - E[i]) + K[i]*dp[1] + 1 - K[i] - E[i]
 * (因为对于叶子节点,没有继续下去的儿子,且G[i].size() = 1)
 * 那么可以发现从叶子节点往上看,每一个儿子节点都可以表示成A*dp[1] + B*dp[father] + C的形式,那么对于一个点的所有子节点
 * 不难发现其father是一样的,并且带入当前的father节点可以算出相应的father节点满足的关系那么从叶节点开始后序遍历这棵树即可最终递推出dp[1]
 * 并且dp[1] = ∑dp[son[1]]/G[1].size()*(1 - 0 - 0) + 0*dp[1] + 1 - 0 - 0
 * 于是dp[1]便可以解出
 * 这里我的代码是将每个节点的A,B,C参数用dp1表示多少倍dp[1], dpfa表示多少倍其父节点,cons表示常数项
 * 递推求解即可
 */

double K[10010], E[10010];
vector  G[10010];
bool vis[10010];
int father[10010];
int n, x, y;

struct Node
{
    double dp1, dpfa, cons;
    Node(double a = 0, double b = 0, double c = 0)
    {
        dp1 = a; dpfa = b; cons = c;
    }
};

Node operator + (Node n1, Node n2)
{
    return Node(n1.dp1 + n2.dp1, n1.dpfa + n2.dpfa, n1.cons + n2.cons);
}

void dfs(int now)
{
    vis[now] = 1;
    for(unsigned int i = 0; i < G[now].size(); i++)
    { 
        if(!vis[G[now][i]])
        {
            father[G[now][i]] = now;
            dfs(G[now][i]);
        }
    }
    return;
}

Node solve(int now)
{
    if(G[now].size() == 1) return Node(K[now], 1 - K[now] - E[now], 1 - K[now] - E[now]);
    Node ret;
    for(unsigned int i = 0; i < G[now].size(); i++)
        if(G[now][i] != father[now])
            ret = ret + solve(G[now][i]);
    double tmp = (1 - ret.dpfa/G[now].size()*(1 - K[now] - E[now]));
    ret.dp1 /= G[now].size(); ret.dp1 *= (1 - K[now] - E[now]); ret.dp1 += K[now];
    ret.dpfa = (1 - K[now] - E[now])/G[now].size();
    ret.cons /= G[now].size(); ret.cons *= (1 - K[now] - E[now]); ret.cons += 1 - K[now] - E[now];
    return Node(ret.dp1/tmp, ret.dpfa/tmp, ret.cons/tmp);
}

int main()
{
    int t;
    scanf("%d", &t);
    for(int cas = 1; cas <= t; cas++)
    {
        scanf("%d", &n);
        for(int i = 1; i <= n; i++) G[i].clear();
        for(int i = 1; i <= n - 1; i++)
        {
            scanf("%d %d", &x, &y);
            G[x].push_back(y);
            G[y].push_back(x);
        }
        for(int i = 1; i <= n; i++)
        {
            scanf("%d %d", &x, &y);
            K[i] = x*1.0/100; E[i] = y*1.0/100;
        }
        memset(vis, 0, sizeof(vis));
        dfs(1);
        Node one;
        for(unsigned int i = 0; i < G[1].size(); i++)
            one = one + solve(G[1][i]);
        if(G[1].size() - one.dp1 - one.dpfa < eps)
        {
            printf("Case %d: impossible\n", cas);
            continue;
        }
        double ans = (one.cons + G[1].size())/(G[1].size() - one.dp1 - one.dpfa);
        if(ans < eps) printf("Case %d: impossible\n", cas);
        else printf("Case %d: %.6f\n", cas, ans);
    }
    return 0;
}


你可能感兴趣的:(ACM_Probability,HDU)