题目链接https://www.acwing.com/problem/content/1214/
思路来源https://www.acwing.com/solution/content/7116/
题目详情:
X 国王有一个地宫宝库,是n×m 个格子的矩阵,每个格子放一件宝贝,每个宝贝贴着价值标签。
地宫的入口在左上角,出口在右下角。
小明被带到地宫的入口,国王要求他只能向右或向下行走。
走过某个格子时,如果那个格子中的宝贝价值比小明手中任意宝贝价值都大,小明就可以拿起它(当然,也可以不拿)。
当小明走到出口时,如果他手中的宝贝恰好是 k 件,则这些宝贝就可以送给小明。
请你帮小明算一算,在给定的局面下,他有多少种不同的行动方案能获得这 k 件宝贝。
输入格式
第一行 3 个整数,n,m,k,含义见题目描述。
接下来 n 行,每行有 m 个整数 Ci 用来描述宝库矩阵每个格子的宝贝价值。
输出格式
输出一个整数,表示正好取 k 个宝贝的行动方案数。
该数字可能很大,输出它对 1000000007取模的结果。
数据范围
1≤n,m≤50
1≤k≤12
0≤Ci≤12输入样例1:
2 2 2 1 2 2 1
输出样例1:
2
输入样例2:
2 3 2 1 2 3 2 1 5
输出样例2:
14
分析:
- 因为数据较小,我们可以采用四层循环
- 又因为每件宝物的价值可能是0,我们初始化时将所有宝物的价值加1,这样数组中下标0变成了边界条件,宝贝价值范围变成了 1~13,这样比较好初始化
- 四层循环:f[i][j][u][v] 表示走到第i行第j列拿了u件物品且最后一件物品的价值为v的方案的数量
- 初始化:f[1][1][1][w[1][1]](第一件取) = f[1][1][0][0](第一件不取)= 1,这两种情况都是一种方案
- 在具体的状态转移中,如果不选,就直接从上一步完完整整地把状态u(选的物品个数)和v(最后一件物品的价值)转移过来;如果选,那上一步就u−1个物品,而且上一步身上所有物品最大值只能是1…v−1,把这些状态的方案数累加,就可以成为选择当前物品的方案数
对应的状态转移方程:
- 情况1:不拿物品
f[i][j][u][v] = (f[i][j][u][v] + f[i-1][j][u][v])%MOD;
f[i][j][u][v] = (f[i][j][u][v] + f[i][j-1][u][v])%MOD;
- 情况2:可以拿物品,需要满足两个条件 (1) u>0,即背包里至少要准备装第一个物品了 (2) v==w[i,j],因为我们的条件就是当前的最后一个物品的价值为v
f[i][j][u][v] = (f[i][j][u][v] + f[i-1][j][u-1][v'])%MOD;
f[i][j][u][v] = (f[i][j][u][v] + f[i][j-1][u-1][v'])%MOD;
代码:
//2022-5-22
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=60,MOD=1000000007;
int n,m,k;
int w[N][N];
int f[N][N][13][14];
int main()
{
cin>>n>>m>>k;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
cin>>w[i][j];
w[i][j]++;
}
//初始化,因为数组下标没有-1,所以让 w[i][j] 都+1
//第一件取,和第一件不取都是一种方案,所以需要初始为 1
f[1][1][1][w[1][1]]=1;
f[1][1][0][0] = 1;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
if(i==1&&j==1) continue; //已经初始化了
for(int u = 0;u<=k;u++)
for(int v = 0;v<=13;v++)//因为对所有物品的价值都加了1,所以最大价值可能是13
{
int &val=f[i][j][u][v]; //用引用,之后书写方便许多
//不拿物品
//因为方案数很多,我们每累加一种情况,就必须取模,防止溢出
val = (val+f[i-1][j][u][v])%MOD;
val = (val+f[i][j-1][u][v])%MOD;
//可以拿物品
if(u>0&&v==w[i][j])
{
//把之前可能的状态数累加,可以得到当前条件得方案数
for(int c=0;c<v;c++)
{
val = (val+f[i-1][j][u-1][c])%MOD;
val = (val+f[i][j-1][u-1][c])%MOD;
}
}
}
}
//当我们拿满了k件物品后,最后一件物品的价值范围是1~13
//将它们累加取余即为答案
int res=0;
for(int i=0;i<=13;i++) res = (res+f[n][m][k][i])%MOD;
cout<<res;
return 0;
}
来自acwing大佬的总结:(大佬好厉害呜呜呜,链接同思路来源)
DP 的细节包括初始化问题,是没有很固定的模板的,一般情况下可以归纳以下三种情形:
1. 求最大值,将所有值初始化为无穷小,找到 DP 的起点(边界),手动赋值;
2. 求最小值,将所有值初始化为无穷大,找到 DP 的起点(边界),手动赋值;
3. 求方案数,将所有值初始化为 00 ,找到 DP 的起点(边界),手动赋值,一般为 11 。
在还没有熟悉的时候,建议认真找到所有边界值,并给它们赋上初值,这样的方式肯定是正确的。
熟悉了以后,针对某些特定的题目,可以根据经验,遵循 不影响最终答案 的原则,省略部分赋值的步骤。