[NOIP2012 普及组] 寻宝（含代码解析）

[NOIP2012 普及组] 寻宝

题目描述

传说很遥远的藏宝楼顶层藏着诱人的宝藏。小明历尽千辛万苦终于找到传说中的这个藏宝楼，藏宝楼的门口竖着一个木板，上面写有几个大字：寻宝说明书。说明书的内容如下：

藏宝楼共有 $N+1$ 层，最上面一层是顶层，顶层有一个房间里面藏着宝藏。除了顶层外，藏宝楼另有 $N$ 层，每层 $M$ 个房间，这 $M$ 个房间围成一圈并按逆时针方向依次编号为 $0,\dots ,M-1$。其中一些房间有通往上一层的楼梯，每层楼的楼梯设计可能不同。每个房间里有一个指示牌，指示牌上有一个数字$x$，表示从这个房间开始按逆时针方向选择第 $x$ 个有楼梯的房间（假定该房间的编号为 $k$），从该房间上楼，上楼后到达上一层的 $k$ 号房间。比如当前房间的指示牌上写着 $2$，则按逆时针方向开始尝试，找到第 $2$ 个有楼梯的房间，从该房间上楼。如果当前房间本身就有楼梯通向上层，该房间作为第一个有楼梯的房间。

寻宝说明书的最后用红色大号字体写着：“寻宝须知：帮助你找到每层上楼房间的指示牌上的数字（即每层第一个进入的房间内指示牌上的数字）总和为打开宝箱的密钥”。

请帮助小明算出这个打开宝箱的密钥。

输入格式

第一行有两个整数 $N$ 和 $M$，之间用一个空格隔开。$N$ 表示除了顶层外藏宝楼共 $N$ 层楼，$M$ 表示除顶层外每层楼有 $M$ 个房间。

接下来 $N\times M$ 行，每行两个整数，之间用一个空格隔开，每行描述一个房间内的情况，其中第 $(i-1)\times M+j$ 行表示第 $i$ 层 $j-1$ 号房间的情况（$i=1,2,\dots ,N$；$j=1,2,\dots ,M$）。第一个整数表示该房间是否有楼梯通往上一层（$0$ 表示没有，$1$ 表示有），第二个整数表示指示牌上的数字。注意，从$j$号房间的楼梯爬到上一层到达的房间一定也是 $j$ 号房间。

最后一行，一个整数，表示小明从藏宝楼底层的几号房间进入开始寻宝（注：房间编号从 $0$ 开始）。

输出格式

一个整数，表示打开宝箱的密钥，这个数可能会很大，请输出对 $20123$ 取模的结果即可。

样例 #1

样例输入 #1

2 3
1 2
0 3
1 4
0 1
1 5
1 2
1

样例输出 #1

5

提示

【数据范围】

对于 $50\%$ 数据，有$0<N\le 1000,0<x\le 10^4$；
对于 $100\%$ 数据，有$0<N\le 10000,0<M\le 100,0<x\le 10^6$。

题目来源

NOIP 2012 普及组 第二题（洛谷）

题解

#include <iostream>
#include <cstdio>
using namespace std;

// 定义两个二维数组 pd 和 a 分别存储房间是否有楼梯和房间的指示牌数字
int pd[10002][202], a[10002][202];

int main() {
    // n 为楼层数，m 为每层的房间数，x 为开始房间号
    int n, m, x, i, j, ans = 0, l = 0;
    scanf("%d%d", &n, &m); // 读取楼层数和每层房间数
    
    // 读取每层房间的信息
    for (i = 1; i <= n; i++) {
        l = 0; // 记录当前层有楼梯的房间数
        for (j = 0; j < m; j++) {
            scanf("%d%d", &pd[i][j], &a[i][j]); // 读取是否有楼梯和指示牌数字
            if (pd[i][j] == 1) l++; // 如果有楼梯，计数加1
        }
        pd[i][m] = l; // 记录当前层有楼梯的房间数到 pd 数组的最后一列
    }
    
    scanf("%d", &x); // 读取开始的房间号
    
    i = 1; // 从第一层开始寻宝
    while (i <= n) {
        ans += a[i][x]; // 将当前房间的指示牌数字加到密钥中
        ans %= 20123; // 对结果取模以避免溢出
        
        int k = 0;
        for (j = x;; j++) {
            if (j == m) j = 0; // 如果到达当前层的最后一个房间，回到第一个房间
            if (pd[i][j] == 1) k++; // 统计有楼梯的房间数
            
            // 找到从当前房间按指示牌数字前进到的目标房间
            if (k == (a[i][x] - 1) % pd[i][m] + 1) break;
        }
        
        x = j; // 更新当前房间号为找到的房间号
        i++; // 进入下一层
    }
    
    printf("%d", ans); // 输出最终的密钥结果
    return 0;
}