1. 问题描述:
农夫约翰对牛棚里昏暗的灯光感到不满,刚刚安装了一个新吊灯。新吊灯由 N 个灯泡组成,这 N 个灯泡围成一圈,编号为 0∼N−1。奶牛对这个新吊灯非常着迷,并且喜欢玩以下游戏:对于第 i 个灯泡,如果在 T−1 时刻,它左侧的灯泡(当 i > 0 时,为第 i−1 个灯泡;当 i=0 时,为第 N−1 个灯泡)是开着,那么在 T 时刻,就切换这个灯泡的状态。这个游戏将持续 B 单位时间。给定灯泡的初始状态,请确定在 B 单位时间后,它们的最终状态。
输入格式
第一行包含两个整数 N 和 B。接下来 N 行,按顺序描述每个灯泡的初始状态,每行包含一个整数 1 (表示开)或 0(表示关)。
输出格式
共 N 行,按顺序每行输出一个灯泡的最终状态。
数据范围
3 ≤ N ≤ 16,
1 ≤ B ≤ 10 ^ 15
输入样例:
5 6
1
0
0
0
0
输出样例:
1
1
1
0
1
样例解释
灯泡状态如下:
时刻 T=0: 1 0 0 0 0
时刻 T=1: 1 1 0 0 0
时刻 T=2: 1 0 1 0 0
时刻 T=3: 1 1 1 1 0
时刻 T=4: 1 0 0 0 1
时刻 T=5: 0 1 0 0 1
时刻 T=6: 1 1 1 0 1
来源:https://www.acwing.com/problem/content/1962/
2. 思路分析:
由题可知可以使用一个二进制串来表示当前n盏灯的状态,0表示不亮,1表示亮,二进制串可以使用字符串或者一个十进制表示的二进制状态数字来存储灯的状态,一般可以使用一个十进制数字来表示这样比字符串来存储状态效率会高一点,由题目可知当前灯的状态只与上一盏灯有关系,若上一盏灯为x,当前灯的状态为y,那么经过上一盏灯的影响之后当前灯的状态为x ^ y,所以我们只需要计算出状态为state情况下的所有灯的状态表示的就是经过一个时刻的所有灯的状态,只需要循环更新B次那么最终就可以得到目标状态,但是这样做的时间复杂度为O(BN) ≈ 16 * 10 ^ 15,肯定会超时所以我们需要优化一下,可以发现所有灯的状态最多有2 ^ 16 = 65536,所以当更新的状态超过65536的时候就一定会重复,当出现了重复之后那么就一定会存在一个环,所以我们可以先一直更新灯的状态如果不存在环但是到达了步数之后那么直接输出答案即可,如果找到了环那么计算还需要走的步数直接更新这么多步数得到的就是目标状态,可以使用数组来记录已经访问过的状态(一开始将所有状态置为-1),值可以记录为到达当前状态的步数,这样当我们发现重复状态的时候使用当前走的步数减去上一次到达当前状态的步数就是环的长度,后面更新对于环的长度取余的步数得到的就是目标状态。
3. 代码如下:
from typing import List
class Solution:
# 输出状态结果
def _print(self, state: int, n: int):
for i in range(n):
print(state >> i & 1)
# 从当前的状态走一步这样可以用来更新当前的状态
def update(self, state: int, n: int):
res = 0
for i in range(n):
# 当前这一位的值
x = state >> i & 1
# 因为是环形的所以需要取余, 第一盏灯的上一盏灯就是最后一盏灯
j = (i - 1 + n) % n
y = state >> j & 1
# 当前灯的状态为上一盏灯的状态与当前灯的状态的异或值, 因为使用十进制数字来表示的所以需要左移i位
res |= (x ^ y) << i
return res
def process(self):
n, m = map(int, input().split())
N = 1 << 16
# 记录对应的状态是否存在这样后面可以用来找环(值为走的步数这样后面可以通过走的步数计算环的长度)
p = [-1] * N
state = 0
for i in range(n):
t = int(input())
# 将第i位点亮
state |= (t << i)
p[state] = 0
i = 1
while True:
# 每走一步更新一下状态
state = self.update(state, n)
if i == m:
# 没有环的情况下输出结果
self._print(state, n)
break
# 之前没有访问过那么就需要记录下来
elif p[state] == -1:
p[state] = i
else:
# 环的长度
_len = i - p[state]
# m - i为剩余的步数对环取余数那么结果就是需要再走的步数
r = (m - i) % _len
# 更新r次
while r > 0:
state = self.update(state, n)
r -= 1
self._print(state, n)
break
i += 1
if __name__ == "__main__":
Solution().process()