中途吃了个饭做了个核酸,压线赛时 AK 了,希望不要挂分。
T1 子集和
题目描述
子集和问题是指,给定 n n n 个数字 a 1 , a 2 , ⋯ , a n a_1,a_2,\cdots, a_n a1,a2,⋯,an,再给定一个目标 t t t,有多少种方法,能够选出一些数字,使得它们的和等于 t t t。注意每个数字可以重复选择多次。
小爱希望计算一些带有限制的子集和问题,她想知道,对于 1 ≤ i ≤ n 1\le i\le n 1≤i≤n,如果规定不能选择 a i a_i ai,那么还有多少种方法,可以选出一些数字,使得它们的和等于目标 t t t?答案对 1 0 9 + 7 10^9+7 109+7 取模。
n ≤ 5000 , a i , t ≤ 1 0 4 n\le 5000, a_i,t\le 10^4 n≤5000,ai,t≤104。
大体思路
对于 1 ≤ i ≤ n 1\le i\le n 1≤i≤n,都需要求出删去第 i i i 个数的方案数,因此考虑动态规划。
记 f [ i , j ] f[i,j] f[i,j] 表示前 i i i 个数之和为 j j j 的方案数, f [ 0 , 0 ] = 1 f[0,0]=1 f[0,0]=1。由于每个数可以重复选取, f [ i , j ] = f [ i − 1 , j ] + f [ i , j − a i ] f[i,j]=f[i-1,j]+f[i,j-a_i] f[i,j]=f[i−1,j]+f[i,j−ai],其中外层循环 i = 1 ∼ n i=1\sim n i=1∼n,内层循环 j = 0 ∼ m j=0\sim m j=0∼m。
同时,记 g [ i , j ] g[i,j] g[i,j] 表示 i i i 开始之后的数之和为 j j j 的方案数, g [ n + 1 , 0 ] = 1 g[n+1,0]=1 g[n+1,0]=1。其状态转移方程与 f f f 类似但 i i i 的方向相反。
容易想到,对于删去 a i a_i ai,相当于前 i − 1 i-1 i−1 个数和 i + 1 i+1