[洛谷] P1161 开灯 Python

题目描述

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

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

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

指定两个数，$a,t$（$a$ 为实数，$t$ 为正整数）。将编号为 $\lfloor a\rfloor ,\lfloor 2\times a\rfloor ,\lfloor 3\times a\rfloor ,\dots ,\lfloor t\times a\rfloor$ 的灯的开关各按一次。其中 $\lfloor k\rfloor$ 表示实数 $k$ 的整数部分。

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

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

3
1.618034 13
2.618034 7
1.000000 21

样例输出 #1

20

提示

记 $T=\sum _{i=1}^n{t}_i=t_1+t_2+t_3+\cdots +t_n$。

• 对于 $30\%$ 的数据，满足 $T\le 1000$；
• 对于 $80\%$ 的数据，满足 $T\le 200000$；
• 对于 $100\%$ 的数据，满足 $T\le 2000000$；
• 对于 $100\%$ 的数据，满足 $n\le 5000$，$1\le a_i<1000$，$1\le t_i\le T$。

数据保证，在经过 $n$ 次操作后，有且只有一盏灯是开的，不必判错。而且对于所有的 $i$ 来说，$t_i\times a_i$ 的最大值不超过 $2000000$。

代码

我一开始尝试用列表来进行反复的添加和删除，虽然结果正确，但是会超时。

后来经过查找发现这种频繁的检查和可能的移除操作在列表中是非常耗时的，因为列表的 remove 方法是一个 O(n) 操作，其中 n 是列表的长度。随着 light 列表的增长，这些操作会变得越来越慢。

import math

"""
第一行输入一个正整数n
接下来n行，每行两个数ai和ti
ai：实数，小数点后有6位
ti：正整数
输出一个正整数，即开灯的编号
"""
# 注：所有的灯一开始都是熄灭的
n = int(input())
ai_list = []	# 创建ai列表
ti_list = []	# 创建ti列表
# 读入数据
for i in range(n):
    ai, ti = map(float, input().split())
    ti = int(ti)
    ai_list.append(ai)
    ti_list.append(ti)

# 创建一个light列表，用来统计亮灯的编号
light = []
for i in range(n):
    for j in range(1, ti_list[i]+1):
        num = math.floor(ai_list[i] * j)	# 计算乘积的向下取整
        # 开关操作
        if num not in light:
            light.append(num)
        else:
            light.remove(num)

print(light[0])

后续我经过改进，发现只要将light列表改为集合即可。

因为集合的查找和删除操作平均时间复杂度为 O(1)。

import math

"""
第一行输入一个正整数n
接下来n行，每行两个数ai和ti
ai：实数，小数点后有6位
ti：正整数
输出一个正整数，即开灯的编号
"""
# 注：所有的灯一开始都是熄灭的
n = int(input())
ai_list = []
ti_list = []
for i in range(n):
    ai, ti = map(float, input().split())
    ti = int(ti)
    ai_list.append(ai)
    ti_list.append(ti)

light = set()	# 将这里更改为集合
for i in range(n):
    for j in range(1, ti_list[i]+1):
        num = math.floor(ai_list[i] * j)
        if num not in light:
            light.add(num)	# 集合的添加是用add函数
        else:
            light.remove(num)

print(list(light)[0])	# 要输出集合的元素需进行list转换

PS：
列表list和集合set的简单测试（转载）：python的列表(list)和集合(set)速度对比，set速度惊人