Codeforces Round #639 (Div. 2) C. Hilbert's Hotel题解

(这题一开始是真的读不懂
这题的题意大致就是:
希尔伯特旅馆内本来已经放满了人,我们输入一个数组,按照这个数组的规则给整个旅馆洗牌。
即对负无穷到正无穷的每一个数x,都把它移动到x+a~x mod n~这个新的位置
问在经过这样一系列操作之后,整个旅馆是否还是一个人占用一个房间?
思路:
旅馆里的所有人是以n为周期地进行处理的。比如cf给的第一个样例"1 14"就是以1为周期的x+k*1(x∈[0,1))都往后移动14个,显然这样仍然能够满足"每个人占用一个房间且没有空房间"
其实如果n=4,那我们只需要让[1,3)里面每个数字经操作过后,都能填到几个周期之后(之前)的位置且模n之后不重复就可以。

#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;

template<class T>inline void read(T &x){x=0;char o,f=1;while(o=getchar(),o<48)if(o==45)f=-f;do x=(x<<3)+(x<<1)+(o^48);while(o=getchar(),o>47);x*=f;}
int cansel_sync=(ios::sync_with_stdio(0),cin.tie(0),0);
#define ll long long
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repb(i,a,b) for(int i=(a);i>=b;i--)
#define INF 0x3f3f3f3f
#define cendl printf("\n");
//#define INF 0x7fffffff

const int MAXN = 2e5+5;
int a[MAXN];
int c[MAXN];
int n,z;

int main(){
	read(z);
	while(z--){
		read(n);
		rep(i,0,n-1){
			read(a[i]);
			c[i]=0;
		}
		rep(i,0,n-1){//检测单个周期n内是否满足条件 
			c[(i+(a[i]%n+n)%n)%n]++;
		}
		bool flag = 1;
		rep(i,0,n-1) if(c[i]!=1){
			flag = 0;break;
		}
		if(flag) cout<<"YES"<<endl;
		else cout<<"NO"<<endl;
	} 
}

你可能感兴趣的:(Codeforces Round #639 (Div. 2) C. Hilbert's Hotel题解)