[USACO09MAR] Cow Frisbee Team S（背包dp）

[USACO09MAR] Cow Frisbee Team S

题目描述

老唐最近迷上了飞盘，约翰想和他一起玩，于是打算从他家的 $N$ 头奶牛中选出一支队伍。

每只奶牛的能力为整数，第 $i$ 头奶牛的能力为 $R_i$。飞盘队的队员数量不能少于 $1$、大于 $N$。一支队伍的总能力就是所有队员能力的总和。

约翰比较迷信，他的幸运数字是 $F$，所以他要求队伍的总能力必须是 $F$ 的倍数。请帮他算一下，符合这个要求的队伍组合有多少？由于这个数字很大，只要输出答案对 $10^8$ 取模的值。

输入格式

第一行：两个用空格分开的整数：$N$ 和 $F$。

第二行到 $N+1$ 行：第 $i+1$ 行有一个整数 $R_i$，表示第 $i$ 头奶牛的能力。

输出格式

第一行：单个整数，表示方案数对 $10^8$ 取模的值。

样例 #1

样例输入 #1

4 5 
1 
2 
8 
2

样例输出 #1

3

提示

对于 $100\%$ 的数据，$1\le N\le 2000$，$1\le F\le 1000$，$1\le R_i\le 10^5$。

思路

• 每只奶牛可以选和不选，总体积必须是F的倍数。
• 此时我们的状态表示：f[i][j]表示 所有已经选好前i个奶牛，且当前的和是j%F的方案数集合。
• 状态转移不就是： f[i][j]=(f[i][j]+f[i-1][j]+f[i-1][((j-w[i])%F+F)%F])%mod;。
  拓展：选和不选这个思路在背包问题很常见，因为我们都得分析最后一个不同点（对于最终结果来说）

代码

#include<iostream>
#include<algorithm>

using namespace std;

const int N = 2010,mod=1e8;

int f[N][N];
int n,F;
int w[N];

int main(){
    cin>>n>>F;
    
    for(int i=1;i<=n;i++)cin>>w[i];
    
    for(int i=1;i<=n;i++)f[i][w[i]%F]=1;
    
    for(int i=1;i<=n;i++){
        for(int j=0;j<F;j++){
            f[i][j]=(f[i][j]+f[i-1][j]+f[i-1][((j-w[i])%F+F)%F])%mod;//选和不选
        }
    }
    
    cout<<f[n][0]%mod;
    
    return 0;
    
}