[报告]HDU 4418 Time travel

Abstract

HDU 4418 Time travel

概率 Markov链 高斯消元

 

Body

Source

http://acm.hdu.edu.cn/showproblem.php?pid=4418

Description

题意不太好懂……

一个人在数轴上来回走，每次以一定概率前进[1, M]步，如果停下来的地方是目标就终止，否则继续该过程直到在目标停下。求走过的路程的期望。

Solution

读明白了就可以发现其实是个比较明显的概率高斯消元题。

首先把(pos,dir)全部展开成一维的坐标，把往回走的方向翻折到另一边即可。这样就有2*(N-1)个点。

令K=2*(N-1)，写出期望方程

e[pos] = sigma{i from 1 to m} (e[(pos+i)%K]+i)*p[i]

边界为e[t]=e[(K-t)%K]=0。

高斯消元解之求e[s]即可。但是tricky的地方就在这里。看以下这组数据：

4 2 2 0 -1

0 100

答案应该为2，而高斯消元解出来的结果是no solution。原因就在于有些状态从s开始走是根本访问不到的，同时也访问不到t，也就是这些状态和整个markov过程根本无关。这些状态之间构成的方程无解就会导致整个解向量不存在，但并不代表e[s]不存在。解决的办法就是将这些状态的期望强制设为INF。判断某个状态能否访问到用bfs就可以了。

Code

#include <iostream>

#include <cassert>

#include <cmath>

#include <cstdio>

#include <cstring>

#include <queue>

#include <algorithm>

using namespace std;



const double EPS = 1e-8;



inline int sgn(double d) {

    if (fabs(d)<EPS) return 0;

    return d>0?1:-1;

}



int gauss(int N, int M, double a[][222]) {

    int i, j, r, c, pvt;

    double maxp;

    for (r=0, c=0; r<N && c<M; ++r, ++c) {

        for (maxp=0, i=r; i < N; ++i)

            if (fabs(a[i][c])>fabs(maxp)) maxp = a[pvt=i][c];

        if (sgn(maxp)==0) {

            r--;

            continue;

        }

        if (pvt != r)

            for (j = r; j <= M; ++j) swap(a[r][j], a[pvt][j]);

        for (j = c+1; j <= M; ++j) {

            a[r][j] /= maxp;

            for (i = r+1; i < N; ++i)

                a[i][j] -= a[i][c]*a[r][j];

        }

    }

    for (i = r; i < N; ++i)

        if (sgn(a[i][M])) return -1;

    if (r < M) return M-r;

    for (i = M-1; i >= 0; --i)

        for (j = i+1; j < M; ++j)

            a[i][M] -= a[j][M]*a[i][j];

    return 0;

}



int T, N, M, s, t, d;

double a[222][222];

double p[111];



int vis[222];



bool bfs(int s) {

    queue<int> q;

    q.push(s);

    memset(vis, 0, sizeof vis);

    vis[s] = 1;

    while (!q.empty()) {

        int u = q.front();

        q.pop();

        for (int i = 0; i < M; ++i) {

            u++; if (u==N) u = 0;

            if (sgn(p[i]) && !vis[u]) {

                vis[u] = 1;

                q.push(u);

            }

        }

    }

    if (vis[t] || vis[(N-t)%N]) return 1;

    else return 0;

}



int main() {

    int i, j, k, r, c;

    cin>>T;

    while (T--) {

        cin>>N>>M>>t>>s>>d;

        assert(N!=0);

        for (i = 0; i < M; ++i) {

            cin>>p[i];

            p[i] /= 100;

        }

        N = N-1<<1;

        if (d > 0) s = N-s;

        if (s==t) {

            puts("0.00");

            continue;

        }

        if (!bfs(s)) {

            puts("Impossible !");

            continue;

        }

        double sum = 0;

        for (i = 0; i < M; ++i)

            sum += p[i]*(i+1);

        memset(a, 0, sizeof a);

        for (i = 0; i < N; ++i) {

            a[i][i] = 1;

            if (!vis[i]) {

                a[i][N] = 1e9;

                continue;

            }

            if (i==t||i==(N-t)%N) {

                a[i][N] = 0;

                continue;

            }

            a[i][N] = sum;

            int now = i;

            for (j = 0; j < M; ++j) {

                now++; if (now==N) now = 0;

                a[i][now] -= p[j];

            }

        }

        if (~gauss(N, N, a)) {

            printf("%.2f\n", a[s][N]+EPS);

        }

        else puts("Impossible !");

    }

    return 0;

}