小A点菜
题目背景
uim 神犇拿到了 uoi 的 ra(镭牌)后,立刻拉着基友小 A 到了一家……餐馆,很低端的那种。
uim 指着墙上的价目表(太低级了没有菜单),说:“随便点”。
题目描述
不过 uim 由于买了一些书,口袋里只剩 M M M 元 ( M ≤ 10000 ) (M \le 10000) (M≤10000)。
餐馆虽低端,但是菜品种类不少,有 N N N 种 ( N ≤ 100 ) (N \le 100) (N≤100),第 i i i 种卖 a i a_i ai 元 ( a i ≤ 1000 ) (a_i \le 1000) (ai≤1000)。由于是很低端的餐馆,所以每种菜只有一份。
小 A 奉行“不把钱吃光不罢休”,所以他点单一定刚好把 uim 身上所有钱花完。他想知道有多少种点菜方法。
由于小 A 肚子太饿,所以最多只能等待 1 1 1 秒。
输入格式
第一行是两个数字,表示 N N N 和 M M M。
第二行起 N N N 个正数 a i a_i ai(可以有相同的数字,每个数字均在 1000 1000 1000 以内)。
输出格式
一个正整数,表示点菜方案数,保证答案的范围在 int 之内。
样例 #1
样例输入 #1
4 4
1 1 2 2
样例输出 #1
3
提示
2020.8.29,增添一组 hack 数据 by @yummy
基本思路
DP + 滚动数组
用f[i][j]表示前 i 道菜用 j 元的方案总数
则 f[i][j] = f[i][j] + f[i-1][j] + f[i-1][j-a[i-1]] ;
再用 滚动数组 优化空间
则 f[j] = f[j] + f[j - a[i]]
Code
#include <bits/stdc++.h>
using namespace std ;
int n , m ;
int a[101] ;
int f[10001] ;
int main ( ) {
cin >> n >> m ;
for ( int i = 1 ;i<=n ; ++i ){
cin >> a[i] ;
}
f[0] = 1 ;//初始化
for ( int i = 1 ; i <= n ; ++i ){//dp
for ( int j = m ; j >= a[i] ; --j ){
f[j] = f[j] + f[j-a[i]] ;//状态转移方程
}
}
cout << f[m] ;//输出答案
return 0 ;
}