洛谷刷题-P1161 开灯——Python题解

题目描述

在一条无限长的路上，有一排无限长的路灯，编号为 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 是正整数。

输出格式

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

输入输出样例

输入 #1

3
1.618034 13
2.618034 7
1.000000 21

输出 #1

20

说明

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

解题心得

看到题目说“无限长”的时候心里咯噔一下，不过说明中对数据量做了限制，不超过2000000。

于是本题采用标记和暴力搜索，用0表示没开灯，1表示开灯，使用循环。

Python解题代码

n = int(input())
lst = [0] * 2000000 # 初始化一个列表lst

for i in range(n):
    at = input().split()
    a, t = float(at[0]), int(at[1])

    for j in range(1, t+1):
        num = int(a* j) # 用int进行取整
        if lst[num] == 0: # 如果关着(0)，就打开(1)
            lst[num] = 1
        else:
            lst[num] = 0 # 如果打开(1)，就关了(0)

    if i == n-1: # 最后一次操作时
        for k in range(len(lst)):
            if lst[k] == 1:
                print(k)
                break

思维发散区

对于倒数第二行的代码中，我一开始写 k+1，是不对的。

但为什么写k，我也有点奇怪，因为路灯是从1开始计数的，列表从0开始，应该差一位才对，希望评论区的大家能帮帮忙。

for k in range(len(lst)): # 将这个变成下面的也是一样print(k)


for k in range(1, len(lst)+1):

同时在洛谷题解区，有大佬给出了高级的异或算法：

https://www.luogu.com.cn/discuss/601906

num = 0
n = int(input())
for i in range(n):
    a, t = input().split()
    t = int(t)
    a = float(a)
    for j in range(1, t + 1):
        num ^= int(a * j)

print(num)

喜欢的同学可以理解一下，思路大概是这样的：

^=表示按位异或操作。

按位异或操作有一个特性：任何数和 0 进行按位异或操作，结果都是它本身；任何数和它自己进行按位异或操作，结果是 0。

  10101010 (这是第一个数)
^ 11001100 (这是第二个数，进行异或)
  --------
  01100110 (这是异或的结果)

对于每次操作，我们计算int(a * j)，这表示 a 乘以 j 的整数部分，然后对这个值进行按位异或操作^。

由于按位异或操作的对称性，除了一个数之外，所有数进行按位异或操作的结果都是 0 。

这意味着，对于每个 a ，我们找到的那个使得按位异或结果不为 0 的数，就是这个 a 对应的操作影响的灯的编号。

在所有操作完成后，num将记录所有这些编号的按位异或结果。由于每个编号最多被按两次（因为灯的状态只有开和关两种），所以最终num中记录的将是所有操作后唯一亮着的灯的编号。

（以上关于异或的解释来自ai）