lightoj 1151 概率dp + 高斯消元

链接：vjudge..

题意：10*10的地图，不过可以直接看成1*100的，从1出发，要到达100，每次走的步数用一个完美的大小为6的骰子决定。地图上有A和B，A和B都使你跳跃，不过一个是往前跳，一个是使你往后跳。。问从1走到100的期望

思路：概率dp的方程很简单就能想到，不过由于可以往前或者往后走，没法直接递推，用高斯消元解方程组得到解。//原来一直都懒得用高斯消元，都想递推解决，不过有些题目用高斯消元确实简单，虽然时间会比递推多

代码：//高斯消元中的无解或者多个解的部分对这道题没有用处，注释掉的solve是输入输出的例子

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

///高斯消元模板
const double eps = 1e-7;    ///精度
const int Max_M = 110;       ///m个方程,n个变量
const int Max_N = 110;
int m, n;
double Aug[Max_M][Max_N+1]; ///增广矩阵
bool free_x[Max_N];         ///判断是否是不确定的变元
double x[Max_N];            ///解集
double fx[110];///记录f（x）

int sign(double x){ return (x>eps) - (x<-eps);}
int Gauss(){
    int row,col,max_r;
    for(row=0,col=0; row<m&&col<n; row++,col++){
        max_r = row;
        for(int i = row+1; i < m; i++){///找到当前列所有行中的最大值(做除法时减小误差)
            if(sign(fabs(Aug[i][col]) - fabs(Aug[max_r][col])) > 0)
                max_r = i;
        }
        if(max_r != row){///将该行与当前行交换
            for(int j = row; j < n+1; j++)
                swap(Aug[max_r][j],Aug[row][j]);
        }
        if(sign(Aug[row][col])==0){///当前列row行以下全为0(包括row行)
            row--;
            continue;
        }
        for(int i = row+1; i < m; i++){
            if(sign(Aug[i][col])==0)
                continue;
            double ta = Aug[i][col]/Aug[row][col];
            for(int j = col; j < n+1; j++)
                Aug[i][j] -= Aug[row][j]*ta;
        }
    }
    ///无解或者多个解的情况
    for(int i = row; i < m; i++){
        if(sign(Aug[i][col]))
            return -1;///col=n存在0...0,a的情况,无解
    }
    if(row < n){
        for(int i = row-1; i >=0; i--){
            int free_num = 0;   ///自由变元的个数
            int free_index;     ///自由变元的序号
            for(int j = 0; j < n; j++){
                if(sign(Aug[i][j])!=0 && free_x[j])
                    free_num++,free_index=j;
            }
            if(free_num > 1) continue;  ///该行中的不确定的变元的个数超过1个,无法求解,它们仍然为不确定的变元
            ///只有一个不确定的变元free_index,可以求解出该变元,且该变元是确定的
            double tmp = Aug[i][n];
            for(int j = 0; j < n; j++){
                if(sign(Aug[i][j])!=0 && j!=free_index)
                    tmp -= Aug[i][j]*x[j];
            }
            x[free_index] = tmp/Aug[i][free_index];
            free_x[free_index] = false;
        }
        return n-row;///存在0...0,0的情况,有多个解,自由变元个数为n-row个
    }
    ///无解或者多个解的情况
    for(int i = n-1; i >= 0; i--){
        double tmp = Aug[i][n];
        for(int j = i+1; j < n; j++)
            if(sign(Aug[i][j])!=0)
                tmp -= Aug[i][j]*x[j];
        x[i] = tmp/Aug[i][i];
    }
    return 0;///有且仅有一个解,严格的上三角矩阵(n==m)
}
/**
返回值:
-1 无解
0 有且仅有一个解
>=1 有多个解,根据free_x判断哪些是不确定的解
**/

/*void solve(){
    cin>>m>>n;///m个方程，n个变量
    memset(Aug,0.0,sizeof(Aug));///矩阵数组
    memset(x,0,sizeof(x));///最终的解的数组
    memset(free_x,1,sizeof(free_x));///判断是否为不确定的变元
    for(int i = 0;i < m;i ++){
        for(int j = 0;j < n;j ++){
            cin>>Aug[i][j];///输入矩阵
        }
    }
    for(int i = 0;i < m;i ++){
        cin>>Aug[i][n];///输入解
    }

    int ans = Gauss();///高斯消元，返回ans代表各种类型的解
    for(int i = 0;i < n;i ++){///输出解
        cout<<x[i]<<endl;
    }
}*/

bool ff[110];
int main()
{
    //freopen("in.txt","r",stdin);
    //solve();
    int T;
    int cas = 1;
    cin>>T;
    n = m = 100;
    while(T --){
        memset(Aug,0.0,sizeof(Aug));
        memset(x,0.0,sizeof(x));
        memset(free_x,1,sizeof(free_x));
        memset(ff,0,sizeof(ff));
        int t;
        cin>>t;
        while(t --){
            int a,b;
            cin>>a>>b;
            ff[a] = 1;
            Aug[a-1][a-1] = 1.0;
            Aug[a-1][b-1] = -1.0;
            Aug[a-1][100] = 0;
        }
        for(int i = 1;i <= 100;i ++){
            if(ff[i] == 0){
                if(i + 6 <= 100){
                    Aug[i-1][i-1] = 1;
                    for(int j = 1;j <= 6;j ++){
                        Aug[i-1][i-1 + j] = -(1.0/6.0);
                    }
                    Aug[i-1][100] = 1;
                }
                else {
                    Aug[i-1][i-1] = 1.0 - (i*1.0 - 94)/6.0;
                    int all = 100 - i;
                    for(int j = 1;j <= all;j ++){
                        Aug[i-1][i-1 + j] = -(1.0/6.0);
                    }
                    Aug[i-1][100] = 1.0;
                }
            }
        }
        Aug[99][99] = 1;Aug[99][100] = 0;
        int a = Gauss();//cout<<a<<endl;
        double ans = x[0];
        printf("Case %d: %.8f\n",cas ++,ans);
    }
    return 0;
}