洛谷 P1192:台阶问题 ← 动态规划 + 前缀和优化

【题目来源】
https://www.luogu.com.cn/problem/P1192

【题目描述】
有 N 级台阶,你一开始在底部,每次可以向上迈 1∼K 级台阶,问到达第 N 级台阶有多少种不同方式。

【输入格式】
两个正整数 N,K。

【输出格式】
一个正整数 ans(mod 100003),为到达第 N 级台阶的不同方式数。

【输入样例】
5 2

【输出样例】
8

【数据范围】
对于 20% 的数据,1≤N≤10,1≤K≤3;
对于 40% 的数据,1≤N≤1000;
对于 100% 的数据,1≤N≤10^5,1≤K≤100。

【算法分析】
● fi 表示从底部到第 i 级台阶的方案数。
● 一维前缀和与差分:https://blog.csdn.net/hnjzsyjyj/article/details/145627134
● 二维前缀和与差分:https://blog.csdn.net/hnjzsyjyj/article/details/145671816

【算法代码】

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;

const int MOD=1e5+3;
const int N=1e5+5;
int f[N],s[N];

int main() {
    int n,k;
    cin>>n>>k;
    f[0]=1,s[0]=1;
    for(int i=1; i<=n; i++) {
        if(i<=k) f[i]=s[i-1]%MOD;
        else {
            f[i]=(s[i-1]-s[i-k-1]+MOD)%MOD;
        }
        s[i]=(s[i-1]+f[i])%MOD;
    }
    cout<<f[n]<<endl;

    return 0;
}

/*
in:5 2
out:8
*/

【参考文献】
https://blog.csdn.net/hnjzsyjyj/article/details/145671816
https://blog.csdn.net/hnjzsyjyj/article/details/145627134
https://blog.csdn.net/hnjzsyjyj/article/details/142732833
https://blog.csdn.net/hnjzsyjyj/article/details/147405964
https://blog.csdn.net/hnjzsyjyj/article/details/145290457
https://blog.csdn.net/hnjzsyjyj/article/details/112797538
https://blog.csdn.net/hnjzsyjyj/article/details/100418268

相关推荐
CQU_JIAKE16 小时前
9.15【A】
算法·动态规划
yi0111 天前
DAY17: LeetCode 139|单词拆分:从“把单词删掉”到用 DP 记录合法切口
人工智能·笔记·python·算法·leetcode·动态规划
老歌老听老掉牙2 天前
动态规划:从递归到最优决策的算法艺术
算法·动态规划
richard_yuu3 天前
动态规划:强化学习的「数学基础」,从 MDP 到值迭代
深度学习·神经网络·算法·yolo·机器学习·动态规划
lvwangshu4 天前
P12960 毒药 题解
动态规划·题解·贪心·技巧·性质题·思维好题
syagain_zsx11 天前
算法基础篇 · 04 前缀和(C++ 题解)
c++·算法·前缀和
彧azz11 天前
算法设计与分析:贪心与动态规划
数据结构·学习·算法·贪心算法·动态规划
Logic10113 天前
C语言/数据结构动态规划题解:Kadane算法求最大子数组和——O(n)时间O(1)空间
c语言·数据结构·动态规划·贪心·时间复杂度·算法题·最大子数组和
梦帮科技13 天前
AI 音乐产品的发布工程:验证门、数据发布、回滚与生产运维纪律
数据结构·数据库·架构·node.js·音视频·动态规划·推荐算法
语戚13 天前
力扣 1621. 大小为K的不重叠线段的数目:动态规划(Java 实现)
java·算法·leetcode·动态规划·力扣·dp