CodeForces 1344 A Hilberts Hotel 数学

最新推荐文章于 2021-01-19 15:57:57 发布

Curren.wong

最新推荐文章于 2021-01-19 15:57:57 发布

阅读量256

点赞数 2

分类专栏：算法刷题笔记 # 数学文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/qq_41729780/article/details/106137102

版权

算法刷题笔记同时被 2 个专栏收录

76 篇文章 3 订阅

订阅专栏

数学

9 篇文章 0 订阅

订阅专栏

文章目录

1. 题目描述

1.1. Limit

Time Limit: 1000 ms

Memory Limit: 131,072 kB

1.2. Problem Description

Hilbert’s Hotel is a very unusual hotel since the number of rooms is infinite! In fact, there is exactly one room for every integer, including zero and negative integers. Even stranger, the hotel is currently at full capacity, meaning there is exactly one guest in every room. The hotel’s manager, David Hilbert himself, decides he wants to shuffle the guests around because he thinks this will create a vacancy (a room without a guest).

For any integer $k$ and positive integer $n$ , let $\mod n$ denote the remainder when $k$ is divided by $n$ . More formally, $\mod n$ is the smallest non-negative integer suchthat $k - r$ is divisible by $n$ . It always holds that $\le k \mod n \le n−1$ . For example, $\mod 12 = 4$ and $\mod 3 = 1$ .

Then the shuffling works as follows. There isan array of $n$ integers $a_0,a_1,\dots,a_{n−1}$ . Then for each integer $k$ , the guest in room $k$ is moved to room number $k+a_k \mod n$ .

After this shuffling process, determine if there is still exactly one guest assigned to each room. That is, there are no vacancies or rooms with multiple guests.

1.3. Input

Each test consists of multiple test cases. The first line contains a single integer $\le t \le 10^4)$ — the number of test cases.

Next $2 t$ lines contain descriptions of test cases.The first line of each test case contains a single integer $(1\le n \le2 \cdot 10^5)$ — the length of the array.

The second line of each test case contains 𝑛n integers $a_0,a_1,\dots,a_{n-1}$ $(−10^9 \le a_i \le 10^9)$ .

It is guaranteed that the sum of $n$ over all test cases does not exceed $\cdot 10^5$ .

1.4. Output

For each test case, output a single line containing “YES” if there is exactly one guest assigned to each room after the shuffling process, or “NO” otherwise. You can print each letter in any case (upper or lower).

1.5. Sample Input

6
1
14
2
1 -1
4
5 5 5 1
3
3 2 1
2
0 1
5
-239 -2 -100 -3 -11

1.6. Sample Output

YES
YES
YES
NO
NO
YES

1.7. Notes

In the first test case, every guest is shifted by $14$ rooms, so the assignment is still unique.

In the second test case, even guests move to the right by $1$ room, and odd guests move to the left by $1$ room. We can show that the assignment is still unique.

In the third test case, every fourth guest moves to the right by $1$ room, and the other guests move to the right by $5$ rooms. We can show that the assignment is still unique.

In the fourth test case, guests $0$ and $1$ are both assigned to room $3$ .

In the fifth test case, guests $1$ and $2$ are both assigned to room $2$ .

1.8. Source

CodeForces 1344 A Hilbert’s Hotel

2. 解读

计算 $a_n + n) \%n + n ) \% n$ ， $\in [0, n - 1]$ ，判断其结果是否有重复，用 set 存储后，比较 set.size() 和 $n$ 的大小即可。

因为 $a_i$ 的范围为 $(−10^9 \le a_i \le 10^9)$ 。

$a_n + n$ 为正数时，结果为 $a_n + n)\%n$ 。
$a_n + n$ 为负数时，结果为 $n + (n + a_n)\%n$ 。

将正负数的情况统一起来，结果为 $a_n + n) \%n + n ) \% n$ 。

3. 代码

#include <iostream>
#include <set>
#include <string.h>
using namespace std;

// 集合存储
set<long long> st;

int main()
{
    // test case
    int t;
    // 个数
    long long n;
    long long buffer;
    scanf("%d", &t);
    // test case
    for (int i = 0; i < t; i++) {
        scanf("%lld", &n);
        // 初始化
        st.clear();
        // 输入
        for (long long j = 0; j < n; j++) {
            scanf("%lld", &buffer);
            st.insert(((j + buffer) % n + n) % n);
        }

        printf("%s\n", (long long)st.size() == n ? "YES" : "NO");
    }
}