【洛谷T89379 【qbxt】复读警告】

题目链接

这个题可以应用dp

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+;
inline int read()
{
    int a=,b=;
    char c=getchar();
    while(c<''||c>'')
    {
        if(c=='-')
            b=-;
        c=getchar();
    }
    while(c>=''&&c<='')
    {
        a=(a<<)+(a<<)+c-'';
        c=getchar();
    }
    return a*b;
}
inline void out(int n)
{
    if(n<)
    {
        putchar('-');
        n=-n;
    }
    if(n>=)
        out(n/);
    putchar(n%+'');
}
int s[];
int dp[][];
int main()
{
    int n=read(),k=read();
    for(int i=;i<=n;i++)
        s[i]=read();
    dp[][]=;
    for(int i=;i<=n;i++)
    {
        for(int j=;j<k;j++)
        {
            int t=j-s[i]%k;
            if(t<)
                t+=k;
            dp[i][j]=(dp[i-][j]+dp[i-][t])%mod;
        }
    }
    out(dp[n][]);
    return ;
}

我们用dp[i][j]表示dp到i，%key为j的方案数

根据推导，我们发现

可以得到这样一个dp转移方程
dp[i][j]=(dp[i-1][j]+dp[i-1][j-s[i]%k])%mod;

保证j-s[i]%k>0

那么可以得到代码