hdu 4326 高斯消元

这篇博客介绍了如何使用高斯消元法解决概率模型问题,具体涉及到一个比赛策略的计算。通过建立概率矩阵并进行消元运算,最终得出队伍中特定位置选手获胜的概率。
摘要由CSDN通过智能技术生成
//dp[i][j]表示当前队伍中第一个人已经赢了i次,当前队伍中第j个人能赢的概率。
//答案就是dp[0][k]
//当j=1时,dp[i][j]=1/4*dp[i+1][j]+3/4*dp[1][n-2]    //该人要么赢,要么输,输的话,后面有两个人排在他后面,所以他在n-2的位置。
//当j=2时,dp[i][j]=1/4*dp[i+1][n-2]+1/4*dp[1][1]+2/4*dp[1][n-1]
//当j=3时,dp[i][j]=1/4*dp[i+1][n-1]+1/4*dp[1][n-1]+1/4*dp[1][1]+1/4*dp[1][n]
//当j=4时,dp[i][j]=1/4*dp[i+1][n]+2/4*dp[1][n]+1/4*dp[1][1];
//当j>4时,dp[i][j]=1/4*dp[i+1][j-3]+3/4*dp[1][j-3]
//另外,dp[m][1]=1,表示队伍中第一个人已经赢了m次,结束。
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cmath>
#include <queue>
#include <vector>
#include <map>
#define pb push_back
#define mp make_pair
#define eps 1e-9
#define zero(x) (fabs(x)<eps)
#define pi acos(-1.0)
#define f1 first
#define f2 second

const int inf=0x3f3f3f3f;
const long long INF=1LL<<50;
using namespace std;
typedef long long LL;
typedef pair <int,double> PII;
#define N 12
#define M 102
int id[N][N];
double a[M][M],x[M];
int n,m,k,gg;
int equ,var;
inline void add(int xx,int y,double p)
{   if (xx==-2) return;
    if (xx==-1)
        x[y]+=p;
    else a[y][xx]-=p;

}
int gauss()
{
    int k,col,max_r;
    for(k=0,col=0;k<equ&&col<var;k++,col++)
    {
        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;
        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][k];
              for(int j=col+1;j<var;j++)a[i][j]-=a[k][j]*a[i][col];
              a[i][col]=0;
          }
    }
    return col==var;
}
void doit()
{   scanf("%d%d%d",&n,&m,&k);
    memset(id,0,sizeof(id));
    for (int i=0;i<m;i++)
    for (int j=1;j<=n;j++)
        id[i][j]=i*n+j-1;
    for (int j=1;j<=n;j++)
    id[m][j]=-2;
    id[m][1]=-1;
    gg=n*m;
    memset(a,0,sizeof(a));
    memset(x,0,sizeof(x));
    for (int i=0;i<gg;i++)
        a[i][i]=1;

    for (int j=1;j<=n;j++)
    {
        if (j==1)
            {   for (int i=0;i<m;i++)
                {add(id[i+1][j],id[i][j],0.25);
                 add(id[1][n-2],id[i][j],0.75);
                }
            }
        if (j==2)
            {   for (int i=0;i<m;i++)
                {add(id[i+1][n-2],id[i][j],0.25);
                 add(id[1][1],id[i][j],0.25);
                 add(id[1][n-1],id[i][j],0.5);
                }
            }
        if (j==3)
            {   for (int i=0;i<m;i++)
                {add(id[i+1][n-1],id[i][j],0.25);
                 add(id[1][n-1],id[i][j],0.25);
                 add(id[1][1],id[i][j],0.25);
                 add(id[1][n],id[i][j],0.25);
                }
            }
        if (j==4)
            {   for (int i=0;i<m;i++)
                {add(id[i+1][n],id[i][j],0.25);
                 add(id[1][n],id[i][j],0.5);
                 add(id[1][1],id[i][j],0.25);
                }
            }
        if (j>4)
            {   for (int i=0;i<m;i++)
                {add(id[i+1][j-3],id[i][j],0.25);
                 add(id[1][j-3],id[i][j],0.75);
                }
            }
    }
    equ=var=gg;

    gauss();
    printf("%lf\n",x[id[0][k]]);
}
int main()
{   int cas,i=0;
    scanf("%d",&cas);
    while (cas--)
        {   printf("Case #%d: ",++i);
            doit();
        }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值