LG5196 「USACO2019JAN」Cow Poetry 背包+乘法原理

\(\mathrm{Cow Poetry}\)

问题描述

LG5196

题解

因为每句诗的长度一定是\(k\)，所以自然而然想到背包。

设\(opt[i][j]\)代表到第\(i\)位时，结尾为\(j\)的方案数。

背包，注意\(\mathrm{DP}\)顺序为先枚举\(i\)，后枚举单词。（Debug了一小时就因为这个）

然后乘法原理统计答案即可。

\(\mathrm{Code}\)

#include<bits/stdc++.h>
using namespace std;

void read(int &x){
    x=0;char ch=1;int fh;
    while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();
    if(ch=='-') ch=getchar(),fh=-1;
    else fh=1;
    while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    x*=fh;
}

void fr(int &x){
    char ch=1;
    while(!(ch>='A'&&ch<='Z')) ch=getchar();
    x=ch-'A'+1;
}

const int mod=1000000007;
const int maxn=5003;
const int maxm=100100;

int n,m,k; 
int s[maxn],c[maxn];
int e[maxn];

int opt[maxn][maxn],sum[maxm];

void Init(){
    read(n);read(m);read(k);
    for(int i=1;i<=n;i++){
        read(s[i]);read(c[i]);
    }
    for(int i=1;i<=m;i++){
        fr(e[i]);
    }
}

void dp(){
    sum[0]=1;
    for(int i=1;i<=k;i++){
        for(int j=1;j<=n;j++){
            if(i<s[j]) continue;
            opt[i][c[j]]=(opt[i][c[j]]+sum[i-s[j]])%mod;
            sum[i]=(sum[i]+sum[i-s[j]])%mod;
        }
    }
}

int tot,cnt[27];

int ksm(long long x,int p){
    long long ret=1;
    while(p){
        if(p&1) ret=ret*x%mod;p>>=1;
        x=x*x%mod;
    }
    return ret;
}

void Work(){
    dp();
    long long ans=0,sum=1;
    for(int i=1;i<=m;i++) ++cnt[e[i]];
    for(int i=1;i<=26;i++){
        if(!cnt[i]) continue;
        ans=0;
        for(int j=1;j<=n;j++){
            if(!opt[k][j]) continue;
            ans=(ans+ksm(opt[k][j],cnt[i]))%mod;
        }
        sum=sum*ans%mod;
    }
    printf("%lld\n",sum);
}

int main(){
    freopen("poetry.in","r",stdin);freopen("poetry.out","w",stdout);
    Init();Work();
    fclose(stdin);fclose(stdout);
    return 0;
}

转载于:https://www.cnblogs.com/liubainian/p/11566462.html