HDU 4418 Time travel 概率 高斯

这篇博客写得不错,我没有思路和debug的时候都是参考这篇博客的，所以题意和思路就不写了。

http://972169909-qq-com.iteye.com/blog/1689107

坑点：m可能大于n呀，没考虑这点，WA到死。

题目地址：

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

/*********************************************
    Problem : HDU 4418
    Author  : NMfloat
    InkTime (c) NM . All Rights Reserved .
********************************************/

#include <map>
#include <set>
#include <queue>
#include <stack>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <vector>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>

#define rep(i,a,b)  for(int i = (a) ; i <= (b) ; i ++) //遍历
#define rrep(i,a,b) for(int i = (b) ; i >= (a) ; i --) //反向遍历
#define repS(it,p) for(auto it = p.begin() ; it != p.end() ; it ++) //遍历一个STL容器
#define repE(p,u) for(Edge * p = G[u].first ; p ; p = p -> next) //遍历u所连接的点
#define cls(a,x)   memset(a,x,sizeof(a))
#define eps 1e-8

using namespace std;

const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
// const int MAXN = 1e5+5;
const int MAXE = 2e5+5;

typedef long long LL;
typedef unsigned long long ULL;

int T,n,m;
int X,Y,D;
// Y 是终点 X 是起点 D 是方向

int fx[] = {0,1,-1,0,0};
int fy[] = {0,0,0,-1,1};

const int MAXN = 205; //最多可建100个方程。
double a[MAXN][MAXN],x[MAXN]; // 矩阵a存的是系数矩阵 矩阵x存的是右边的值
int equ ; // 方程数 和 未知数个数

void debug() {
    rep(i,1,equ) {
        rep(j,1,equ) {
            printf("%f ",a[i][j]);
        }
        printf("= %f\n",x[i]);
    }
    puts("");
}

bool gauss() {// 0代表无解，1代表有解
    // debug();
    rep(i,1,equ) { // 代表equ次矩阵变换
        //用第i行第i个数消除第i列。
        //在第i列中找到最大的一个数 从第i行到第equ行
        int maxd = i; //maxd代表第几行 
        rep(j,i+1,equ) if(fabs(a[maxd][i])<fabs(a[j][i])) maxd = j;
        if(fabs(a[maxd][i])<eps) return false; //无解
        //如果最大的数在i行，就不用交换，否则就要交换行。
        if(maxd != i) {
            //交换 maxd 行和 i 行
            rep(j,i,equ) swap(a[maxd][j],a[i][j]);
            swap(x[maxd],x[i]);
        }

        //用i行i列的数去消除其他行
        rep(j,i+1,equ) { //代表现在消除的是哪一行
            double rate = a[j][i] / a[i][i];
            rep(k,i,equ) {//代表第j行第k个数 
                a[j][k] -= rate * a[i][k];
            }
            x[j] -= rate * x[i];
        }
        //debug();
    }
    // debug();
    //消元结束，现在开始找出结果
    rrep(i,1,equ) {
        rep(j,i+1,equ) {
            x[i] -= a[i][j] * x[j];
        }
        x[i] /= a[i][i]; 
    }
    // debug();
    return true;
    //x数组存的就是最后结果。
}


//0 1 2 3 2 1 0特殊的编号方式
bool vis[205];
int  idx[205];//重新编号
double rate[205];
double sum;

//E[Y] = 0;
//E[x] = (E[x+1] + 1) * rate[1] + (E[x+2] + 2) * rate[2] + ... + (E[x+m] + m) * rate[m];
queue<int>q;

bool BFS() {
    cls(vis,0); cls(a,0); cls(x,0);
    while(!q.empty()) q.pop();
    if(D == 1) X = (n -  X) % n; //根据方向确定X的位置
    q.push(X);
    idx[X] = 1; vis[X] = true; equ = 0;
    int count = 1;
    bool flg = false;

    while(!q.empty()) {
        int tmp = q.front(); q.pop();
        ++ equ;
        if(tmp == Y || tmp == n - Y) {
            a[equ][idx[tmp]] = 1;
            x[equ] = 0;
            flg = true;
            continue;
        }
        x[equ] = -sum;
        a[equ][idx[tmp]] = -1; //这个一定要在前面,因为后面也许还会有到达a[equ][idx[tmp]]的要加上去
        rep(i,1,m) {
            int pos = ( tmp + i) % n; // 找到下一个位置 因为位置是0 1 2 3 2 1
            if(fabs(rate[i]) < eps) continue;
            if(!vis[pos]) {
                vis[pos] = true;
                idx[pos] = ++count;
                //printf("pos : %d\n",pos);
                q.push(pos);
            }
            a[equ][idx[pos]] += rate[i]; //这个是因为也许m比n要大,然后转一圈转到了相同的位置.
        }

    }
    return flg;
}

void input() {
    scanf("%d %d %d %d %d",&n,&m,&Y,&X,&D);
    sum = 0;
    rep(i,1,m) { 
        scanf("%lf",&rate[i]); rate[i] /= 100.0;
        sum += rate[i] * i;
    }   
}

void solve() {
    n = 2 * n - 2;
    if(!BFS()) {
        //永远达不到Y节点。
        puts("Impossible !");
    }
    else {
        // puts("Hello");
        if(!gauss()) {puts("Impossible !");}
        else printf("%.2f\n",x[1]);
    }
}

int main(void) {
    //freopen("a.in","r",stdin);
    scanf("%d",&T); while(T--) {
        input();
        solve();
    }
    return 0;
}