【洛谷 P1990】覆盖墙壁 题解(动态规划)

覆盖墙壁

题目描述

你有一个长为 N N N 宽为 2 2 2 的墙壁,给你两种砖头:一个长 2 2 2 宽 1 1 1,另一个是 L 型覆盖 3 3 3 个单元的砖头。如下图:

复制代码
0  0
0  00

砖头可以旋转,两种砖头可以无限制提供。你的任务是计算用这两种来覆盖 N × 2 N\times 2 N×2 的墙壁的覆盖方法。例如一个 2 × 3 2\times3 2×3 的墙可以有 5 5 5 种覆盖方法,如下:

复制代码
012 002 011 001 011  
012 112 022 011 001

注意可以使用两种砖头混合起来覆盖,如 2 × 4 2\times4 2×4 的墙可以这样覆盖:

复制代码
0112
0012

给定 N N N,要求计算 2 × N 2\times N 2×N 的墙壁的覆盖方法。由于结果很大,所以只要求输出最后 4 4 4 位。例如 2 × 13 2\times 13 2×13 的覆盖方法为 13465 13465 13465,只需输出 3465 3465 3465 即可。如果答案少于 4 4 4 位,就直接输出就可以,不用加前导 0 0 0,如 N = 3 N=3 N=3 时输出 5 5 5。

输入格式

一个整数 N N N,表示墙壁的长。

输出格式

输出覆盖方法的最后 4 4 4 位,如果不足 4 4 4 位就输出整个答案。

样例 #1

样例输入 #1

复制代码
13

样例输出 #1

复制代码
3465

提示

数据保证, 1 ≤ N ≤ 1000000 1\leq N\leq 1000000 1≤N≤1000000。


思路

f[i] 表示铺满前 i 列的方法数。

g[N] 表示铺满前 i 列,第 i+1 列只铺一格的方法数。

作图法求状态转移方程。

对于f[x]

f[i - 1]

f[i - 2]

g[i - 3]

有两种情况。

对于g[x]

以 g[i - 3] 为例。

f[i - 3]

g[i - 3]

cpp 复制代码
g[i - 2] = f[i - 3] + g[i - 3];

化简得

cpp 复制代码
g[i] = f[i - 1] + g[i - 1];

状态转移方程:

cpp 复制代码
f[i] = f[i - 1] + f[i - 2] + 2 * g[i - 2];
g[i] = f[i - 1] + g[i - 1];

用列举法不难得到初始情况下f[1]、f[2]、g[1]和g[2]的值。

使用 %10000 来保证结果只包含最后4位。

注意:每次计算动态转移方程后需要立即取模,否则WA。


AC代码

cpp 复制代码
#include <iostream>
#define AUTHOR "HEX9CF"
using namespace std;

const int N = 1e7 + 5;
const int M = 10000;

// 铺满前i列
int f[N];
// 铺满前i列,第i+1列只铺一格
int g[N];

int main()
{
    int n;
    cin >> n;
    f[1] = 1;
    f[2] = 2;
    g[1] = 1;
    g[2] = 2;
    for (int i = 3; i <= n; i++)
    {
        f[i] = (f[i - 1] + f[i - 2] + 2 * g[i - 2]) % M;
        g[i] = (f[i - 1] + g[i - 1]) % M;
    }
    cout << f[n] % M << endl;
    return 0;
}
相关推荐
BB_CC_DD24 分钟前
四. 以Annoy算法建树的方式聚类清洗图像数据集,一次建树,无限次聚类搜索,提升聚类搜索效率。(附完整代码)
深度学习·算法·聚类
YHY_13s28 分钟前
访问者模式
c++·访问者模式
我也不曾来过11 小时前
list底层原理
数据结构·c++·list
A charmer1 小时前
C++ 日志系统实战第三步:熟悉掌握各种设计模式
c++·日志系统
Ethon_王1 小时前
STL容器适配器详解:queue篇
c++
静听夜半雨1 小时前
CANoe入门——3、新建LIN工程及LIN DataBase(LDF文件)的创建
网络·数据库·c++·编辑器
梁下轻语的秋缘2 小时前
每日c/c++题 备战蓝桥杯 ([洛谷 P1226] 快速幂求模题解)
c++·算法·蓝桥杯
CODE_RabbitV2 小时前
【深度强化学习 DRL 快速实践】逆向强化学习算法 (IRL)
算法
虾球xz2 小时前
游戏引擎学习第244天: 完成异步纹理下载
c++·学习·游戏引擎
矛取矛求2 小时前
C++区别于C语言的提升用法(万字总结)
c语言·c++