链家笔试1

最新推荐文章于 2021-11-15 16:06:07 发布

weixin_30872671

最新推荐文章于 2021-11-15 16:06:07 发布

阅读量69

点赞数

原文链接：http://www.cnblogs.com/2390624885a/p/7468015.html

版权

链家笔试1

题目描述
　　小明喜欢在火车旅行的时候用手机听音乐，他有N首歌在手机里，在整个火车途中，他可以听P首歌，所以他想产生一个播放表产生P首歌曲，这个播放表的规则是：

·　　(1)每首歌都要至少被播放一次
·　　(2)在两首一样的歌中间，至少有M首其他的歌

小明在想有多少种不同的播放表可以产生，那么给你N,M,P，你来算一下，输出结果取1000000007的余数

输入：

　　输入N,M,P
　　N范围在1到100
　　M范围在0到N
　　P范围在N到100

输出：

　　输出结果mod 1000000007的余数

样例输入
1 0 3
样例输出
1
提示
其他样例

1 1 3

2 0 3

50 5 100

222288991

题解
记录到了第i个位置，已经用了j种颜色的方案数，我们不需要考虑第i个位置要放哪种歌，而是考虑能放几种不同种类的歌。记忆化搜索即可。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#define ll long long
#define mo 1000000007
using namespace std;
ll dp[105][105];
int n,m,p;
ll dfs(int i,int j)
{
    if (dp[i][j]!=-1) return dp[i][j];
    if (i==p)
    {
        if (j==n){dp[i][j]=1;return 1;}
        else{dp[i][j]=0;return 0;}  
    }
    dp[i][j]=0;
    if (j>m)dp[i][j]=dfs(i+1,j)*(j-m);
    if (j<n)dp[i][j]+=dfs(i+1,j+1)*(n-j);
    if (dp[i][j]>=mo) dp[i][j]%=mo;
    return dp[i][j];
}
int main()
{
    scanf("%d%d%d",&n,&m,&p);
    for (int i=0;i<=p;i++)
        for (int j=0;j<=n;j++)
            dp[i][j]=-1;
    printf("%lld",dfs(0,0));
    return 0;
}