XTU 1233 Coins(DP)

Coins

Accepted : 95 Submit : 243
Time Limit : 1000 MS Memory Limit : 65536 KB

Coins

Problem Description:

Duoxida buys a bottle of MaiDong from a vending machine and the machine give her n coins back. She places them in a line randomly showing head face or tail face on. And Duoxida wants to know how many situations that m continuous coins head face on among all possible situations. Two situations are considered different if and only if there is at least one position that the coins' faces are different.

Input

The first line contains a integer T(no more than 20) which represents the number of test cases.

In each test case, a line contains two integers n and m.()

Output

For each test case, output the result modulo  in one line.

Sample Input

2
4 2
5 2

Sample Output

8
19

题意:投掷n次硬币,问有连续m个正面向上的情况有多少种

思路:本来以为是组合问题,可是愣是没想出来公式。 于是想到了DP,可是由于自己写的时候状态写错了TLE,后来看了别人的题解才看懂。

很是巧妙。 定义数组dp[i][2],dp[i][0]表示投掷i次硬币满足连续m次正面向上的情况数,dp[i][1]表示投掷i次硬币不满足连续m次正面向上的情况数,也就是说,dp[i][0]+dp[i][1]等于投掷i次硬币的情况总数。  而我们知道,投i次硬币的情况总数a[i]是2的i次方(每投一次有正和反两个状态)。

当i<m时,不可能有连续m次正面向上的情况,所以dp[i][0]=0,dp[i][1]=a[i]

当i==m时,恰好有一种情况可以满足,dp[i][0]=1,dp[i][1]=a[i]-1

当i>m,状态转移方程就是dp[i][0]=2*dp[i-1][0]+dp[i-m-1][1](i-1时就已经满足有连续m次正面向上的情况,那么第i个硬币可正可反,所以乘以2,还有一种情况是在第i个硬币投出正面从而恰好有连续m次正面向上的情况,那么也就是说从第i-m+1到第i个硬币一共m个硬币都是正面,这个时候必须明确第i-m个硬币必须是反面,想一想,为什么?

假设第i-m个硬币也是正面,那么i-m到i-1这m个硬币都是正面,这种情况是在dp[i-1][0]里面的(i-1时已经满足有连续m次正面向上的情况)

所以投掷第i个硬币刚好满足条件的情况数是dp[i-m-1][1])

dp[i][1]=(a[i]-dp[i][0]+mod)%mod注意必须加上mod,负数取模会出错。

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define N 1000010
#define ll long long
#define mod 1000000007
ll dp[N][2];
ll a[N];
int main()
{
    int T;
    int n,m;
    a[0]=1;
    for(int i=1; i<=N; i++)
        a[i]=(2*a[i-1])%mod;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d %d",&n,&m);
        memset(dp,0,sizeof(dp));
        for(int i=0; i<m; i++)
            dp[i][1]=a[i];
        dp[m][0]=1;
        dp[m][1]=(a[m]-dp[m][0]+mod)%mod;
        for(int i=m+1; i<=n; i++)
        {
            dp[i][0]=(2*dp[i-1][0]+dp[i-1-m][1])%mod;
            dp[i][1]=(a[i]-dp[i][0]+mod)%mod;
        }
        printf("%lld\n",dp[n][0]);
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值