XTU 1233 Coins（DP）

最新推荐文章于 2021-12-05 23:08:15 发布

acm_cxq

最新推荐文章于 2021-12-05 23:08:15 发布

阅读量682

点赞数

分类专栏：动态规划文章标签： ACM XTU

本文链接：https://blog.csdn.net/acm_cxq/article/details/51556157

版权

动态规划专栏收录该内容

42 篇文章 0 订阅

订阅专栏

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

 
 CoinsProblem 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.
InputThe 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.()
OutputFor each test case, output the result modulo  in one line.
Sample Input2
 4 2
 5 2
 
Sample Output8
 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;
}