http://acm.hdu.edu.cn/showproblem.php?pid=1204
糖果大战
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Problem Description
生日Party结束的那天晚上,剩下了一些糖果,Gandon想把所有的都统统拿走,Speakless于是说:“可以是可以,不过我们来玩24点,你不是已经拿到了一些糖果了吗?这样,如果谁赢一局,就拿走对方一颗糖,直到拿完对方所有的糖为止。”如果谁能算出来而对方算不出来,谁就赢,但是如果双方都能算出或者都不能,就算平局,不会有任何糖果的得失。
Speakless是个喜欢提前想问题的人,既然他发起了这场糖果大战,就自然很想赢啦(不然可就要精光了-_-)。现在他需要你的帮忙,给你他每局赢的概率和Gardon每局赢的概率,请你给出他可能获得这场大战胜利的概率。
Speakless是个喜欢提前想问题的人,既然他发起了这场糖果大战,就自然很想赢啦(不然可就要精光了-_-)。现在他需要你的帮忙,给你他每局赢的概率和Gardon每局赢的概率,请你给出他可能获得这场大战胜利的概率。
Input
每行有四个数,Speakless手上的糖果数N、Gardon手上的糖果数M(0<=N,M<=50)、一局Speakless能解答出来的概率p、一个问题Gardon能解答出来的概率q(0<=p,q<=1)。
Output
每行一个数,表示Speakless能赢的概率(用百分比计算,保留到小数点后2位)。
Sample Input
50 50 0.5 0.510 10 0.51 0.550 50 0.51 0.5
Sample Output
0.50
0.60
0.88
题意:
A和B玩游戏一开始分别有n颗和m颗糖果..然后玩一个游戏,.赢的几率分别是p和q..若一方赢了而且另一方输了,则赢的一方就从对方手上拿一颗糖果..问A把B所有糖果赢过来的几率是多少..
大致思路:
几个星期前才在某大牛的教导下学会了这类题目的做法..
设f[i]为A手上有i颗糖果最后赢的几率, 那么
a. A能得到一颗糖果的几率P=p(1-q)/(p(1-q)+q(1-p)); 对应地B的几率Q=q(1-p)/(q(1-p)+p(1-q))
b. f[n+m] = 1, f[0] = 0, f[i]=Qf[i-1]+Pf[i+1] ==> Qf[i-1]+Pf[i+1]-f[i]=0 ( 1<i<n+m )
那么以f为未知数就有n+m+1个非其次方程组,用消元法解方程后f[n]就是答案.
代码:
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define MAXN 110
#define eps (1e-8)
#define INF 1000000000
#define abs(x) ( (x) > 0? (x): -(x) )
#define sqr(x) ((x) * (x))
#define MAX(a, b) ((a) > (b)? (a): (b))
#define MIN(a, b) ((a) < (b)? (a): (b))
typedef long long LL;
int n, m;
double p, q, pp, qq, f[MAXN], mtx[MAXN][MAXN];
void swap( double &x, double &y ) { double temp = x; x = y; y = temp; }
void solve( double mtx[][MAXN], int size )
{
for ( int i = 0; i <= size; ++i )
{
if ( abs( mtx[i][i] ) < eps )
{
for ( int j = i + 1; j <= size; ++j ) if ( abs( mtx[j][i] ) > eps )
{
for ( int k = i; k <= size + 1; ++k )
swap( mtx[j][k], mtx[i][k] );
break;
}
}
if ( abs( mtx[i][i] ) > eps )
{
double c = mtx[i][i];
for ( int k = i; k <= size + 1; ++k ) mtx[i][k] /= c;
for ( int j = i + 1; j <= size; ++j ) if ( mtx[j][i] > eps )
{
c = mtx[j][i];
for ( int k = i; k <= size + 1; ++k ) mtx[j][k] /= c;
for ( int k = i; k <= size + 1; ++k ) mtx[j][k] -= mtx[i][k];
}
}
}
/*
for ( int i = 0; i < size; ++i )
{
for ( int j = 0; j < size; ++j ) cout << mtx[i][j] << ' ';cout << endl;
}
*/
for ( int i = size; i >= 0; --i ) if ( abs( mtx[i][i] ) > eps )
{
for ( int j = i - 1; j >= 0; --j ) if ( abs( mtx[j][i] ) > eps )
{
mtx[j][size + 1] -= mtx[i][size + 1] * mtx[j][i];
mtx[j][i] = 0;
}
}
}
int main()
{
while ( scanf( "%d%d%lf%lf", &n, &m, &p, &q ) != EOF )
{
if ( n == 0 )
puts( "0.00" );
else if ( m == 0 )
puts( "1.00" );
else if ( p < eps )
puts( "0.00" );
else if ( q < eps )
puts( "1.00" );
else if ( q > 1 - eps )
puts( "0.00" );
else if ( p > 1 - eps )
puts( "1.00" );
else
{
int t = n + m;
pp = ( 1 - q ) * p / ( ( 1 - q ) * p + ( 1 - p ) * q );
qq = 1 - pp;
memset( mtx, 0, sizeof(mtx) );
for ( int i = 1; i < t; ++i )
{
mtx[i][i - 1] = qq;
mtx[i][i + 1] = pp;
mtx[i][i] = -1;
}
//cout << " " << pp << endl;
mtx[t][t] = mtx[t][t + 1] = 1;
mtx[0][0] = 1; mtx[0][t + 1] = 0;
solve( mtx, t );
printf( "%.2lf\n", mtx[n][t + 1] / mtx[n][n] );
}
}
return 0;
}