牛客練習賽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;
}