HDU 5819 Knights(概率DP)

23 篇文章 0 订阅

Description
数轴1~n每点有一个骑士,每个骑士有个初始方向di(0表示向右,1表示向左1),骑士单位时间移动单位长度,两个骑士相遇会决斗,每个骑士赢的概率是0.5,输的骑士挂掉,骑士到达0和n+1时会改变方向,问最后活下来的那位骑士是第n位骑士的概率
Input
第一行一整数T表示用例组数,每组用例首先输入一整数b表示骑士数量,之后n个01表示每个骑士的初始方向(T<=10,1<=n<=1000)
Output
对于每组用例,输出最后活下来的那位骑士是第n位骑士的概率,结果模1e9+7
Sample Input
2
2
0 0
3
0 1 0
Sample Output
Case #1: 500000004
Case #2: 250000002
Solution
第一个骑士往右走,第n个骑士往左走,第n个骑士活下来的情况就是他打败了他左边所有往右走的骑士,令dp[i][j]表示第i名骑士往右走且前i名骑士中有j名骑士方向向右的概率,那么有以下转移方程
若第i名骑士方向向右,那么有dp[i][j]=dp[i-1][j-1]
若第i名骑士方向向左,设前i-1个骑士中k个骑士方向向右,考虑这个骑士往左走打败k-j个骑士后被干掉,那么有这里写图片描述,j=2,3,…,i-1
注意到j=1时有两种情况,一种是第i名骑士干掉前面一些方向向右的骑士后被最后一个向右的骑士干掉,另一种是第i名骑士干掉前面所有向右的骑士后到0点改变方向,故这里写图片描述
进一步优化这个转移方程有
dp[i][1]=dp[i-1][1]+dp[i][2]
dp[i][j]=(dp[i-1][j]+dp[i][j+1])/2,j=2,3,..,i-1
最后答案就是dp[n][1]/2,除以2是因为最后向右的必须是第n个骑士,也就是说第n个骑士与其左边最后一个骑士决斗要胜出,而不是被打败
Code

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
#define mod 1000000007ll
#define inv2 500000004ll
#define maxn 1111
ll dp[maxn][maxn];
int T,n,d,Case=1;
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        printf("Case #%d: ",Case++);
        scanf("%d",&n);
        memset(dp,0,sizeof(dp));
        dp[0][0]=1;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&d);
            if(i==1)d=1;
            if(i==n)d=0;
            if(d)
            {
                for(int j=1;j<=i;j++)
                    dp[i][j]=dp[i-1][j-1];
            }
            else
            {
                for(int j=i-1;j>1;j--)
                    dp[i][j]=(dp[i-1][j]+dp[i][j+1])%mod*inv2%mod;
                dp[i][1]=(dp[i-1][1]+dp[i][2])%mod;
            }
        }
        printf("%I64d\n",dp[n][1]*inv2%mod);
    }
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值