AcWing 1960. 闪烁

题目链接

1960. 闪烁 - AcWing题库

题目描述

农夫约翰对牛棚里昏暗的灯光感到不满，刚刚安装了一个新吊灯。

新吊灯由 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≤1015
输入样例：
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

思路： 

对于本题，所有灯泡的状态一共有2^16种，但是题目给的状态变化次数最大为1e15次，所以中间肯定有某些状态围成了一个环，因为对于每一个灯泡，他左边的位置是1的话，他的状态就需要发生改变，为了方便对于样例1 0 0 0 0， 不妨记为0 0 0 0 1，每次改变只需要看一下他右边的状态是不是1，如果是的话就改变即可，那么如果过程中出现了环，我们就可以对于这个环进行遍历即可

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1<<16;

int n,m;
int p[N];

int update(int state) {
	int res=0;
	for(int i=0; i<n; i++) {
		int j=(i-1+n)%n;//每一位右边的一位
		int x=state>>i&1,y=state>>j&1;
		res|=(x^y)<<i;
	}
	return res;
}

void print(int state) {
	for(int i=0; i<n; i++) {
		cout<<(state>>i&1)<<endl;
	}
}

signed main() {
	cin>>n>>m;

	int state=0;
	for(int i=0; i<n; i++) {
		int x;
		cin>>x;
		state|=x<<i;
	}
//	cout<<state<<endl;

	memset(p,-1,sizeof p);
	p[state]=0;

	for(int i=1;; i++) {
		state=update(state);
		if(i==m) {
			print(state);
			break;
		} else if(p[state]==-1) {
			p[state]=i;
		} else {
			int len=i-p[state];
			int r=(m-i)%len;
			while(r--) {
				state=update(state);
			}
			print(state);
			break;
		}
	}
}