L25.【LeetCode笔记】 三步问题的四种解法(含矩阵精彩解法!)

news2025/1/3 16:24:13

目录

1.题目

2.三种常规解法

方法1:递归做

​编辑

方法2:改用循环做

初写的代码

提交结果

分析

修改后的代码

提交结果

for循环的其他写法

提交结果

方法3:循环+数组

提交结果

3.方法4:矩阵

算法

代码实践

1.先计算矩阵n次方

2.后将矩阵n次方嵌入递推式中

提交结果


1.题目

https://leetcode.cn/problems/three-steps-problem-lcci/

三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。

示例1:

 输入:n = 3 
 输出:4
 说明: 有四种走法

示例2:

 输入:n = 5
 输出:13

提示:

  1. n范围在[1, 1000000]之间

2.三种常规解法

方法1:递归做

和之前青蛙跳台阶的思想一样(参见35.【C语言】详解函数递归文章),先找递推公式,再写递归

int recursion(int n)
{
    if (n==1)
        return 1;
    if (n==2)
        return 2;
    if (n==3)
        return 4;
    return (recursion(n-1)+recursion(n-2)+recursion(n-3))%1000000007;
}
int waysToStep(int n) 
{
    return recursion(n);
}

算法上没问题,但是时间复杂度过高,提交后没有通过

方法2:改用循环做

初写的代码

int waysToStep(int n) 
{

    if (n==1)
        return 1;
    
    if (n==2)
        return 2;
        
    if (n==3)
        return 4; 

    int a=1;
    int b=2;
    int c=4;
    int d=0;
    for (int i=3;i<n;i++)
    {
        d=a+b+c;
        a=b;
        b=c;
        c=d;
    }
    return c%1000000007;
}

提交结果

分析

虽然代码中返回值写成c%1000000007,但是没有完全领会题目的意思,c的值并没有真正改变,可以看看报错的数字:当n==61时,"2082876103 + 1748130326"相加溢出了,可以设想2082876103和1748130326产生的原因,n==某个数溢出了,可以使程序溢出的n的临界值

将代码最后改成return c;测试n的值

多次尝试后

当未模1000000007时,

n==34n==35n==34
61569347411324368522082876103

615693474+1132436852=1748130326(大于1000000007),求出了出错提示上的两个数字

a+b+c可能数值超过int的范围,因此要分两次模1000000007,由于d=a+b+c,则程序的计算顺序为:先算a+b,后算+c,则应该对(a+b)先模1000000007再+c,再对d模一次

修改后的代码

        d=(a+b)%1000000007+c;
        d%=1000000007;
        a=b;
        b=c;
        c=d;
提交结果

for循环的其他写法


    for (int i=3;i<n;i++)
    {
        d=(a+b)%1000000007+c;
        a=b;
        b=c;
        c=d;
        c%=1000000007;
    }
提交结果

方法3:循环+数组

int waysToStep(int n) 
{
    if (n==1)
        return 1;
    if (n==2)
        return 2;
    if (n==3)
        return 4;

    int* arr=(int*)malloc(sizeof(int)*(n+1));
    arr[1]=1;
    arr[2]=2;
    arr[3]=4;
    for (int i=4;i<=n;i++)
    {
        arr[i]=(arr[i-3]+arr[i-2])%1000000007+arr[i-1];
        arr[i]%=1000000007;
    }
    return arr[n];

}
提交结果

3.方法4:矩阵

算法

F(n)=F(n-1)+F(n-2)+F(n-3)

改写成矩阵形式

F(n)=\begin{bmatrix} 1 & 1 & 1 \end{bmatrix}\begin{bmatrix} F(n-1)\\ F(n-2)\\ F(n-3)\\ \end{bmatrix}

F(n-1)=\begin{bmatrix} 1 & 0 & 0 \end{bmatrix}\begin{bmatrix} F(n-1)\\ F(n-2)\\ F(n-3)\\ \end{bmatrix}

F(n-2)=\begin{bmatrix} 0 & 1 & 0 \end{bmatrix} \begin{bmatrix} F(n-1)\\ F(n-2)\\ F(n-3) \end{bmatrix}

将上方三个式子合三为一

\begin{bmatrix} F(n)\\ F(n-1)\\ F(n-2) \end{bmatrix}=\begin{bmatrix} 1 & 1 & 1\\ 1 &0 & 0\\ 0 & 1 & 0 \end{bmatrix}\begin{bmatrix} F(n-1)\\ F(n-2)\\ F(n-3) \end{bmatrix}(关键式子)

递推

\begin{bmatrix} F(n-1)\\ F(n-2)\\ F(n-3) \end{bmatrix}=\begin{bmatrix} 1 & 1 & 1\\ 1 &0 & 0\\ 0 & 1 & 0 \end{bmatrix}^2\begin{bmatrix} F(n-2)\\ F(n-3)\\ F(n-4) \end{bmatrix}

\begin{bmatrix} F(n-2)\\ F(n-3)\\ F(n-4) \end{bmatrix}=\begin{bmatrix} 1 & 1 & 1\\ 1 &0 & 0\\ 0 & 1 & 0 \end{bmatrix}^3\begin{bmatrix} F(n-3)\\ F(n-4)\\ F(n-5) \end{bmatrix}......

可以一直递推到

**************************************************************************************************************

\begin{bmatrix} F(n)\\ F(n-1)\\ F(n-2) \end{bmatrix}=\begin{bmatrix} 1 & 1 & 1\\ 1 &0 & 0\\ 0 & 1 & 0 \end{bmatrix}^{n-3}\begin{bmatrix} F(3)\\ F(2)\\ F(1) \end{bmatrix}

**************************************************************************************************************  

\begin{bmatrix} 1& 1 &0 \\ 1& 0 &0 \\ 0& 1&0 \end{bmatrix}^{n-3}=\begin{bmatrix} a & b &c \\ d& e&f\\ g& h & i \end{bmatrix}则最终答案为F(n)=a*F(3)+b*F(2)+cF(1)=4a+2b+c

代码实践

1.先计算矩阵n次方\begin{bmatrix} 1& 1 &0 \\ 1& 0 & 0\\ 0&1 & 0 \end{bmatrix}^n

//矩阵[1,1,1;1,0,0;0,1,0]的n次方(n为计算次数)
#define _CRT_SECURE_NO_WARNINGS
#include <stdlib.h>
#include <stdio.h>
int main()
{
	int arr1[3][3] = { 1,1,1,1,0,0,0,1,0 };
	int arr2[3][3] = { 0 };
	int arr3[3][3] = { 1,1,1,1,0,0,0,1,0 };
	int n;
	scanf("%d", &n);
	for (int i = 1; i <= n; i++)
	{
		if (i % 2)//i为奇数
		{
			for (int i = 0; i < 3; i++)
			{
				for (int j = 0; j < 3; j++)
				{
					arr2[i][j] = 0;
					for (int k = 0; k < 3; k++)
					{
						arr2[i][j] += arr3[i][k] * arr1[k][j];
					}
				}
			}
		}
		else
		{
			for (int i = 0; i < 3; i++)
			{
				for (int j = 0; j < 3; j++)
				{
					arr3[i][j] = 0;
					for (int k = 0; k < 3; k++)
					{

						arr3[i][j] += arr2[i][k] * arr1[k][j];
					}
				}
			}
		}
	}

	if (n % 2)
	{
		for (int i = 0; i < 3; i++)
		{
			for (int j = 0; j < 3; j++)
			{
				printf("%d ", arr2[i][j]);
			}
			printf("\n");
		}
	}
	else
	{
		for (int i = 0; i < 3; i++)
		{
			for (int j = 0; j < 3; j++)
			{
				printf("%d ", arr3[i][j]);
			}
			printf("\n");
		}
	}
	return 0;
}

2.后将矩阵n次方嵌入递推式中

int waysToStep(int n) 
{
    if (n==1)
        return 1;
    if (n==2)
        return 2;
    if (n==3)
        return 4;

    long long  arr1[3][3] = { 1,1,1,1,0,0,0,1,0 };
	long long  arr2[3][3] = { 0 };
	long long  arr3[3][3] = { 1,1,1,1,0,0,0,1,0 };
	n-=4;//不是-3,计算的是矩阵n次方的运行次数
	for (int i = 1; i <= n; i++)
	{
		if (i % 2)//i为奇数
		{
			for (int i = 0; i < 3; i++)
			{
				for (int j = 0; j < 3; j++)
				{
					arr2[i][j] = 0;
					for (int k = 0; k < 3; k++)
					{
						arr2[i][j] += (arr3[i][k] * arr1[k][j])%1000000007;
					}
				}
			}
		}
		else
		{
			for (int i = 0; i < 3; i++)
			{
				for (int j = 0; j < 3; j++)
				{
					arr3[i][j] = 0;
					for (int k = 0; k < 3; k++)
					{

						arr3[i][j] += (arr2[i][k] * arr1[k][j])%1000000007;
					}
				}
			}
		}
	}
    if (n%2)
        return (arr2[0][0]*4+arr2[0][1]*2+arr2[0][2])%1000000007;
    else
        return (arr3[0][0]*4+arr3[0][1]*2+arr3[0][2])%1000000007;
}

提交结果

封装成函数 

其实封装成函数代码看起来更简洁

void calc_matirx_power(long long int (*a)[3] ,long long int (*b)[3] ,long long int (*c)[3] )
{
    for (int i = 0; i < 3; i++)
			{
				for (int j = 0; j < 3; j++)
				{
					a[i][j] = 0;
					for (int k = 0; k < 3; k++)
					{
						a[i][j] += (b[i][k] * c[k][j])%1000000007;
					}
				}
			}
}

int waysToStep(int n) 
{
    if (n==1)
        return 1;
    if (n==2)
        return 2;
    if (n==3)
        return 4;
 
    long long  arr1[3][3] = { 1,1,1,1,0,0,0,1,0 };
	long long  arr2[3][3] = { 0 };
	long long  arr3[3][3] = { 1,1,1,1,0,0,0,1,0 };
	n-=4;//不是-3,计算的是矩阵n次方的运行次数
	for (int i = 1; i <= n; i++)
	{
		if (i % 2)//i为奇数
		{
			calc_matirx_power(arr2,arr3,arr1);
		}
		else
		{
            calc_matirx_power(arr3,arr2,arr1);
		}
	}
    if (n%2)
        return (arr2[0][0]*4+arr2[0][1]*2+arr2[0][2])%1000000007;
    else
        return (arr3[0][0]*4+arr3[0][1]*2+arr3[0][2])%1000000007;
}

注意calc_matrix_power参数类型的写法:long long int (*a)[3]

这种写法可以看看这篇文章:★♛★指针(重难点)合集

提交结果

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2269324.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

小白投资理财 - 看懂 PE Ratio 市盈率

小白投资理财 - 看懂 PE Ratio 市盈率 什么是 PE RatioPE 缺陷PE 优点总结 无论是在菜市还是股票市场&#xff0c;每个人都想捡便宜&#xff0c;而买股票就像市场买菜&#xff0c;必须货比三家&#xff0c;投资股票最重要就是要知道回本时间要多久&#xff0c;市场上很多时候散…

python利用selenium实现大麦网抢票

大麦网&#xff08;damai.cn&#xff09;是中国领先的现场娱乐票务平台&#xff0c;涵盖演唱会、音乐会、话剧、歌剧、体育赛事等多种门票销售。由于其平台上经常会有热门演出&#xff0c;抢票成为许多用户关注的焦点。然而&#xff0c;由于票务资源的有限性&#xff0c;以及大…

SQL-leetcode-183. 从不订购的客户

183. 从不订购的客户 Customers 表&#xff1a; -------------------- | Column Name | Type | -------------------- | id | int | | name | varchar | -------------------- 在 SQL 中&#xff0c;id 是该表的主键。 该表的每一行都表示客户的 ID 和名称。 Orders 表&#…

充电桩语音提示芯片方案-支持平台自定义语音NV128H让充电更便捷

随着工业化与城市化进程的加速推进&#xff0c;传统燃油汽车的数量急剧攀升&#xff0c;这直接导致了石油资源的过度开采与消耗。石油&#xff0c;作为不可再生的化石燃料&#xff0c;其储量日益枯竭&#xff0c;价格波动频繁&#xff0c;给全球能源安全带来了前所未有的挑战。…

Android 旋转盘导航栏

1.直接上源码&#xff1a; package com.you.arc;import android.content.Context; import android.graphics.Canvas; import android.graphics.Color; import android.graphics.Paint; import android.graphics.Point; import android.graphics.RectF; import android.support…

手搓一个ChatUI需要分几步

只关注项目代码的同学可以直接跳转目录中的正文部分&#xff0c;查看项目仓库和功能介绍。 引言 Chatbot的UI界面设计&#xff0c;是和传统软件不同的&#xff1a;都是当面一个简洁的对话框&#xff0c;框里预备着热乎的工具&#xff0c;可以随时更新。 像我这样做工的牛马&a…

低代码开发深度剖析:JNPF 如何引领变革

在当今数字化转型加速的时代&#xff0c;低代码开发已成为众多企业提升效率、降低成本的关键利器。它打破了传统开发模式的高门槛和冗长流程&#xff0c;让应用开发变得更加高效、灵活与普惠。 低代码开发的核心优势 低代码开发平台通过可视化的操作界面&#xff0c;减少了对…

uniapp实现APP、小程序与webview页面间通讯

需求&#xff1a; 1、需要在Uniapp开发的APP或小程序页面嵌入一个H5网页&#xff0c;需要拿到H5给APP传递的数据。 2、并且这个H5是使用vuevant开发的。&#xff08;其实跟使用uniapp开发H5一样&#xff09; 实现步骤&#xff1a; 1、首先需要兼容多端和App端&#xff0c;因…

网络游戏之害

网络游戏之害&#xff1a; 网络游戏于今之世风靡四方&#xff0c;其娱人耳目、畅人心怀之效&#xff0c;固为人知&#xff0c;然所藏之害&#xff0c;若隐伏之暗潮&#xff0c;汹涌而至时&#xff0c;足以覆舟&#xff0c;尤以青年为甚&#xff0c;今且缕析其害&#xff0c;以…

数据库的创建与删除:理论与实践

title: 数据库的创建与删除:理论与实践 date: 2024/12/31 updated: 2024/12/31 author: cmdragon excerpt: 在当今的数字时代,数据的管理和存储变得尤为重要。数据库作为数据存储的结构化方案,为数据的增删改查提供了系统化的方法。在一个典型的数据库管理系统中,创建和…

【电机控制】基于STC8H1K28的六步换向——方波驱动(软件篇)

【电机控制】基于STC8H1K28的六步换向——方波驱动&#xff08;软件篇&#xff09; 文章目录 [TOC](文章目录) 前言一、main.c二、GPIO.c三、PWMA.c四、ADC.c五、CMP.c六、Timer.c七、PMSM.c八、参考资料总结 前言 【电机控制】STC8H无感方波驱动—反电动势过零检测六步换向法 …

优化租赁小程序提升服务效率与用户体验的策略与实践

内容概要 在这个快速发展的商业环境中&#xff0c;租赁小程序成为了提升服务效率和用户体验的重要工具。通过对用户需求的深入挖掘&#xff0c;我们发现他们对于功能的便捷性、响应速度和界面的友好性有着极高的期待。因此&#xff0c;针对这些需求&#xff0c;完善租赁小程序…

电子电器架构 --- 智能座舱与AI结合

我是穿拖鞋的汉子&#xff0c;魔都中坚持长期主义的汽车电子工程师。 老规矩&#xff0c;分享一段喜欢的文字&#xff0c;避免自己成为高知识低文化的工程师&#xff1a; 所谓鸡汤&#xff0c;要么蛊惑你认命&#xff0c;要么怂恿你拼命&#xff0c;但都是回避问题的根源&…

C语言 斐波那契数列(递归算法)

C语言 斐波那契数列&#xff08;递归算法&#xff09; 什么是斐波那契数列&#xff1f; 是指这样一个数列&#xff1a;0&#xff0c;1&#xff0c;1&#xff0c;2&#xff0c;3&#xff0c;5&#xff0c;8&#xff0c;13……这个数列从第3项开始 &#xff0c;每一项都等于前两…

互联网直播点播平台EasyDSS无人机视频推拉流技术实现工地远程监控巡检直播

在建筑行业&#xff0c;施工现场的安全管理和实时监控一直是项目管理中的重点。随着技术的进步&#xff0c;无人机工地直播技术成为了一种新兴的解决方案&#xff0c;它不仅能够提高施工透明度&#xff0c;还能够加强现场安全管理。EasyDSS作为一种先进的流媒体技术平台&#x…

【每日学点鸿蒙知识】模拟器开启网络、长时任务、兼容性测试支持、丢帧定位、SO中访问rawfile等

1、模拟器如何开启网络&#xff1f; 模拟器使用的是电脑本身的网络&#xff0c;不通过代理即可访问网络。 2、创建子window后&#xff0c;锁屏很短时间内&#xff0c;应用会被杀死&#xff1f; 没开长时任务&#xff0c;锁屏和退后台保活要开长时任务。 应用退至后台后&…

计算机网络 (18)使用广播信道的数据链路层

一、广播信道的基本概念 广播信道是一种允许一个发送者向多个接收者发送数据的通信信道。在计算机网络中&#xff0c;广播信道通常用于局域网&#xff08;LAN&#xff09;内部的主机之间的通信。这种通信方式的主要优点是可以节省线路&#xff0c;实现资源共享。 二、广播信道数…

深度学习——损失函数汇总

1. 连续值损失函数 总结:主要使用胡贝儿损失函数,应用于连续数值的预测之间的误差损失,参考地址 import torch import torch.nn as nna = torch.tensor([[1, 2], [3, 4]], dtype=torch.float) b = torch.tensor([[3, 5], [8, 6]], dtype=torch.float)loss_fn1 = torch.nn.M…

github 项目分享

今天和大家分享一些github上面搜到关于卫星遥感和水环境相关的项目。 一、WaterDetect 使用端到端算法去识别水体范围的算法&#xff0c;针对哨兵2卫星遥感数据可用。 项目地址&#xff1a; https://github.com/cordmaur/WaterDetect 二、DeepWaterMap 深度卷积神经网络去…

音视频入门基础:MPEG2-PS专题(2)——使用FFmpeg命令生成ps文件

一、错误的命令 通过FFmpeg命令可以将mp4文件转换为ps文件&#xff0c;PS文件中包含PS流数据。 由于PS流/PS文件对应的FFInputFormat结构为&#xff1a; const FFInputFormat ff_mpegps_demuxer {.p.name "mpeg",.p.long_name NULL_IF_CONFIG_SMALL…