动态规划--摆花(二维dp)
#include <iostream> using namespace std; //dp[i][j]表示第i种花位置,第j个位置为止 long long int dp[120][120]; long long int a[160]; int main() { int n,m; cin>>n>>m;//n种花m盆 for(int i=1;i<=n;i++){ cin>>a[i]; } dp[0][0]=1; for(int i=1;i<=n;i++ ){ for(int j=0;j<=m;j++){ for(int k=0;k<=j&&k<=a[i];k++){ dp[i][j]=(dp[i][j]+dp[i-1][j-k])%(1000007); } } } cout<<dp[n][m]; // 请在此输入您的代码 return 0; }
方法:二维动态规划
思路是:
利用一个数组dp[i][j]表示到第i种花第j个位置为止
花种的下标从1开始
dp[0][0]=1 -->到第0种花第0个位置初始化为1
核心代码:
for(int i=1;i<=n;i++ ){ for(int j=0;j<=m;j++){ for(int k=0;k<=j&&k<=a[i];k++){ dp[i][j]=(dp[i][j]+dp[i-1][j-k])%(1000007); } } }
花种的下标为1
遍历n种花,j个位置0~m(因为初始化dp[0][0]为1,dp[i-1][j-k]下标从[0][0]开始)
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
· .NET周刊【3月第1期 2025-03-02】