HDU 4870 Rating 2014 Multi-University Training Contest 1 J题 概率DP+高斯消元

题目大意：

就是现在有一个比赛,初始Rating是0, 每次参加比赛有P的概率Rating变为 Min(Rating + 50, 1000) 有(1 - P)的概率变为 Max(0, Rating -100)

为现在一个人用两个账号,每次选择用其中Rating较低的号参加比赛, 使得其中1个号的Rating达到1000所需要比赛的期望场数

大致思路：

很明显的概率DP, 由于变量数不多可以用高斯消元

不过这题的误差问题表示有些不能理解, Gauss消元误差较大

代码如下：

Result  :  Accepted     Memory  :  1812 KB     Time  :  1794 ms

/*
* Author: Gatevin
* Created Time: 2015/1/29 14:14:54
* File Name: Iris_Fleyja.cpp
*/
#include<iostream>
#include<sstream>
#include<fstream>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<ctime>
#include<iomanip>
using namespace std;
const double eps(1e-8);
typedef long long lint;

double a[300][300];
double x[300];
int hash[22][22];
int var, equ;

/*
* hash[i][j]为E[i][j]对应的变量编号
* 最终解的结果存在x[hash[i][j]]中
*/

/*
* 如果用E[i][j] (i >= j)表示当前分数分别为i和j时到达目标状态的概率
* E[i][j] = P*(1 + E[i][j + 1]) + (1 - P)*(1 + E[i][Max(j - 2, 0)]) (i >= j)
* 很明显E[i][j] = E[j][i], 所以高斯消元做的话只需要一半的变量就够了
* 另外感觉这题误差还是有些....不好说..
*/

int Gauss()//正常的高斯消元的想法
{
for(int k = 0, col = 0; k < equ && col < var; k++, col++)
{
int max_r = k;
for(int i = k + 1; i < equ; i++)
if(fabs(a[i][col]) > fabs(a[max_r][col]))
max_r = i;
//if(fabs(a[max_r][col]) < eps) return 0;
/*
* 对于eps 为1e-8精度不够,方程组会被判无解
* 注释掉之后莫名过了...
*/
if(k != max_r)
{
for(int j = col; j < var; j++)
swap(a[k][j], a[max_r][j]);
swap(x[k], x[max_r]);
}
x[k] /= a[k][col];
for(int j = col + 1; j < var; j++) a[k][j] /= a[k][col];
a[k][col] = 1;
for(int i = 0; i < equ; i++)
if(i != k)
{
x[i] -= x[k]*a[i][col];
for(int j = col + 1; j < var; j++) a[i][j] -= a[k][j]*a[i][col];
a[i][col] = 0;
}
}
return 1;
}

void buildEquation(double P)//建立方程组
{
memset(a, 0, sizeof(a));
memset(x, 0, sizeof(x));
var = equ = 0;
memset(hash, -1, sizeof(hash));
for(int i = 0; i <= 20; i++)
{
hash[20][i] = var++;
x[equ] = 0;
a[equ++][hash[20][i]] += 1;
}
for(int i = 0; i < 20; i++)
for(int j = 0; j <= i; j++)
{
if(hash[i][j] == -1)
hash[i][j] = var++;
if(j + 1 <= i && hash[i][j + 1] == -1)
hash[i][j + 1] = var++;
if(j + 1 > i && hash[j + 1][i] == -1)
hash[j + 1][i] = var++;
if(j < 2)
{
if(hash[i][0] == -1)
hash[i][0] = var++;
a[equ][hash[i][0]] -= (1 - P);
a[equ][hash[i][j]] += 1;
if(j + 1 <= i)
a[equ][hash[i][j + 1]] -= P;
else
a[equ][hash[j + 1][i]] -= P;
x[equ++] = 1;
}
else
{
if(hash[i][j - 2] == -1)
hash[i][j - 2] = var++;
a[equ][hash[i][j - 2]] -= (1 - P);
a[equ][hash[i][j]] += 1;
if(j + 1 <= i)
a[equ][hash[i][j + 1]] -= P;
else
a[equ][hash[j + 1][i]] -= P;
x[equ++] = 1;
}
}
return;
}

int main()
{
double p;
while(scanf("%lf", &p) != EOF)
{
buildEquation(p);
Gauss();
printf("%.6f\n", x[hash[0][0]]);
}
return 0;
}