门|门模拟 2017 | 第 49 题

该题是一道数字模拟题，要求实现一个小球在多个门之间运动的过程，并输出最终停留的门编号。

题目描述

在一个长为n的走廊上有若干个门，第i个门有一个长度为a[i]的区间，小球每次可以从一个门跳到相邻的门或者停留在原地，每个门只能通过一次，小球初始位于第一个门的区间内，问最终小球停留在哪一个门的区间内。

输入格式：

• 第一行包含整数n(1 <= n <= 100)，表示门的数量。
• 第二行包含n个整数，分别表示每个门的区间长度$a[i]$。(1 <= a[i] <= 100)

输出格式：

• 输出一个整数，表示最终停留在哪一个门的区间内。

样例输入输出

输入样例

5
2 3 1 4 5

输出样例

5

解题思路

我们可以通过模拟小球每一步的移动，最终求得小球停留的门编号。

具体实现为，用一个变量pos记录小球当前所在的门编号，然后循环移动小球，直到小球无法移动位置。每次移动，我们首先看小球所在的门的区间是否有相邻门，如果有，我们移动到相邻门，否则就停留在原地。同时，每次移动后，我们标记走过的门，直到所有门都被标记为止。

最后输出停留的门编号即可。

代码实现

def main():
    n = int(input().strip())
    a = list(map(int, input().strip().split()))
    
    # 记录每个门是否被访问过
    vis = [False] * n
    
    # 小球只能在已访问的区间中移动
    pos = 0
    vis[0] = True
    while True:
        # 记录可以移动到的门的索引
        next_pos = []
        if pos > 0 and not vis[pos-1]:
            next_pos.append(pos-1)
        if pos < n-1 and not vis[pos+1]:
            next_pos.append(pos+1)
        if not next_pos:
            break
        pos = next_pos[0]
        vis[pos] = True
    
    print(pos+1)

代码中用vis数组记录每个门是否被访问过，用pos表示当前小球所在的门的索引。当一个门被访问时，我们将对该门所对应的数组vis进行标记，并将当前门的索引变为已访问过的门中的一个门的索引。在循环中，我们每次只要找出可以移动到的门的索引，然后随机一个门移动到即可。如果所有门都已访问过，循环结束时，小球所在的门的索引即为最终停留的门的索引加1，输出即可。

代码片段如上所示，返回markdown格式。