Cow Poetry——读题+dp

Description

在这里插入图片描述

Input

在这里插入图片描述

Output

在这里插入图片描述

Sample Input

3 3 10
3 1
4 1
3 2
A
B
A
Sample Output

960
Hint

在这个例子中,Bessie认识三个单词。前两个单词押韵,长度分别为三个音节和四个音节,最后一个单词长度为三个音节,不与其他单词押韵。她想要写一首三行的诗,每行包含十个音节,并且第一行和最后一行押韵。共有960首这样的诗。以下是一首满足要求的诗(其中1,2、3分别代表第一个、第二个、第三个单词):121 123 321。

题意:

告诉你n个单词的长度和韵部,让你组成m*k的矩阵,接下来告诉你m行韵部,不同行相同的韵部必须相同,不同的韵部可以相同!,问你有多少种情况。

题解:

假设 A 模式韵脚取 i , 则情况数为 dp[i]^num[A],那么A 模 式 所 有 可 能 的 情 况 就 是
dp[1]^ num[A]+dp[2]^ num[A]+dp[3]^ num[A]+…+dp[n]^num[A]
dp就用完全背包就好了

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll mod=1e9+7;
const int N=5e3+5;
ll dp[N],cnt[N];
ll len[N],c[N];
ll qpow(ll a,ll b)
{
    ll ret=a,ans=1;
    while(b)
    {
        if(b&1)
            ans=ans*ret%mod;
        ret=ret*ret%mod;
        b>>=1;
    }
    return ans;
}
ll num[105];
int main()
{
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++)
        scanf("%lld%lld",&len[i],&c[i]);
    char s[2];
    for(int i=1;i<=m;i++)
        scanf("%s",s),num[s[0]-'A']++;
    dp[0]=1;
    for(int i=1;i<k;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(len[j]>i)
                continue;
            (dp[i]+=dp[i-len[j]])%=mod;
        }
    }
    for(int i=1;i<=n;i++)
        (cnt[c[i]]+=dp[k-len[i]])%=mod;
    ll ans=1;
    for(int i=0;i<26;i++)
    {
        if(num[i]==0)
            continue;
        ll sum=0;
        for(int j=1;j<=n;j++)
            if(cnt[j])
                sum=(sum+qpow(cnt[j],num[i]))%mod;
        (ans*=sum)%=mod;
    }
    printf("%lld\n",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/tianyizhicheng/article/details/88135360