【Python入门】​P1161开灯​ & enumerate 的基本用法

P1161

开灯

题目描述

在一条无限长的路上，有一排无限长的路灯，编号为 1,2,3,4,…。

每一盏灯只有两种可能的状态，开或者关。如果按一下某一盏灯的开关，那么这盏灯的状态将发生改变。如果原来是开，将变成关。如果原来是关，将变成开。

在刚开始的时候，所有的灯都是关的。小明每次可以进行如下的操作：

指定两个数，a,t（a 为实数，t 为正整数）。将编号为 ⌊a⌋,⌊2×a⌋,⌊3×a⌋,…,⌊t×a⌋ 的灯的开关各按一次。其中 ⌊k⌋ 表示实数 k 的整数部分。

在小明进行了 n 次操作后，小明突然发现，这个时候只有一盏灯是开的，小明很想知道这盏灯的编号，可是这盏灯离小明太远了，小明看不清编号是多少。

幸好，小明还记得之前的 n 次操作。于是小明找到了你，你能帮他计算出这盏开着的灯的编号吗？

输入格式

第一行一个正整数 n，表示 n 次操作。

接下来有 n 行，每行两个数，ai​,ti​。其中 ai​ 是实数，小数点后一定有 6 位，ti​ 是正整数。

输出格式

仅一个正整数，那盏开着的灯的编号。

import math

# 读取输入的整数 n，表示 n 次操作
n = int(input())

# 初始化一个数组来记录每盏灯被按开关的次数
max_light_id = 2000000 + 1  # 确保数组足够大
light_count = [0] * (max_light_id + 1)

# 循环读取接下来的 n 行
for _ in range(n):
    # 读取每行的输入，并将其拆分为 a 和 t
    a, t = input().split()
    
    # 将 a 转换为浮点数，t 转换为整数
    a = float(a)
    t = int(t)
    
    # 遍历 k 从 1 到 t，计算灯的编号并更新 light_count
    for k in range(1, t + 1):
        light_id = math.floor(k * a)
        if light_id <= max_light_id:  # 防止索引越界
            light_count[light_id] += 1

# 遍历 light_count，找到被按了奇数次开关的灯
for light_id, count in enumerate(light_count):
    if count % 2 == 1:
        print(light_id)
        break

1. enumerate 的基本用法

语法：

enumerate(iterable, start=0)

• iterable：需要遍历的可迭代对象（如列表、元组、字符串等）。

• start：索引的起始值，默认为 0。

返回值：

enumerate 返回一个枚举对象，每次迭代时生成一个元组 (index, value)，其中：

• index 是当前元素的索引。

• value 是当前元素的值。

2. for light_id, count in enumerate(light_count): 的含义

 light_count 是一个列表，存储了每盏灯被按开关的次数。例如：

light_count = [0, 1, 0, 2, 1]

• enumerate(light_count) 会生成以下枚举对象:

• [(0, 0), (1, 1), (2, 0), (3, 2), (4, 1)]

  • 0 是索引，对应 light_count[0] 的值 0。

  • 1 是索引，对应 light_count[1] 的值 1。

  • 以此类推。

• 其中：

• for light_id, count in enumerate(light_count): 会依次遍历这些元组，并将索引赋值给 light_id，值赋值给 count。

3.传统写法 vs enumerate 写法

传统写法：

for i in range(len(light_count)):
    light_id = i
    count = light_count[i]
    print(f"light_id = {light_id}, count = {count}")

enumerate 写法：

for light_id, count in enumerate(light_count):
    print(f"light_id = {light_id}, count = {count}")