牛客练习赛50 B tokitsukaze and Hash Table 并查集

链接：https://ac.nowcoder.com/acm/contest/1080/B
来源：牛客网
 

题目描述

tokitsukaze有n个数，需要按顺序把他们插入哈希表中，哈希表的位置为0到n-1。

插入的规则是：
刚开始哈希表是空的。
对于一个数x，在哈希表中，如果(x mod n)的位置是空的，就把x放在(x mod n)的位置上。如果不是空的，就从(x mod n)往右开始找到第一个空的位置插入。若一直到n-1都不是空的，就从位置0开始继续往右找第一个空的位置插入。

因为哈希表总共有n个空位，需要插入n个数，所以每个数都能被插入。

现在tokitsukaze想知道把这n个数按顺序插入哈希表后，哈希表中的每个位置分别对应的是哪个数。

输入描述:

第一行包含一个正整数n(1≤n≤10^6)。
第二行包含n个非负整数x(0≤x≤10^9)，这些数按从左到右的顺序依次插入哈希表。

输出描述:

输出一行，n个数，第i个数表示哈希表中位置为i所对应的数。(0≤i≤n-1)

示例1

输入

4
1 2 6 5

输出

5 1 2 6

说明

插入1时，1 mod 4=1，是空的，在位置1插入。
插入2时，2 mod 4=2，是空的，在位置2插入。
插入6时，6 mod 4=2，不是空的，找到下一个空的位置为3，所以在位置3插入。
插入5时，5 mod 4=1，不是空的，找到下一个空的位置为0，所以在位置0插入。

示例2

输入

4
3 0 7 11

输出

0 7 11 3

说明

插入3时，3 mod 4=3，是空的，在位置3插入。
插入0时，0 mod 4=0，是空的，在位置0插入。
插入7时，7 mod 4=3，不是空的，找到下一个空的位置为1，所以在位置1插入。
插入11时，11 mod 4=3，不是空的，找到下一个空的位置为2，所以在位置2插入。

题解：

解法一：
用并查集维护空位。在位置x插入一个数后，合并x和(x+1)%n即可。
要注意，合并的时候，(x+1)%n必须当爹，才能达到维护的效果。

解法二：
用set维护所有空位，每次lower_bound找到第一个可用空位即可。(可能会卡常)

我用set写的代码，连续交了5发，有一发超时，set的时间是并查集的两三倍。

下边给出两种代码：

并查集代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
int father[maxn];
int ans[maxn];
int get(int x){
    return x==father[x]?x:father[x]=get(father[x]);
}
int main(){
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++) father[i]=i;
    for(int i=1;i<=n;i++){
        int x;
        scanf("%d",&x);
        int y=x%n;
        int f=get(y);
        ans[f]=x;
        int z=get((f+1)%n);
        father[f]=z;
    }
    for(int i=0;i<n;i++) printf("%d ",ans[i]);
    return 0;
}

set代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+5;
int a[maxn];
set<int> s;
set<int>::iterator it;
int main(){
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;i++) s.insert(i);
    for(int i=0;i<n;i++){
        int x;
        scanf("%d",&x);
        int y=x%n;
        it=s.lower_bound(y);
        if(it==s.end()) it=s.begin();
        a[*it]=x;
        s.erase(it);
    }
    for(int i=0;i<n;i++) printf("%d ",a[i]);
    return 0;
}