#牛客网 牛客练习赛50 B-tokitsukaze and Hash Table （并查集 or set查找）

 

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 524288K，其他语言1048576K
64bit IO Format: %lld

题目描述

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插入。

题目大意 ： 输入N个数 M， 另T = M % N， 如果一个数组中T位置是空缺的，将M放入该位置， 否则一直往后放， 如果后面都没有，则跳到开头开始放， 最后输出放完后的数组

思路 ： 并查集写法 ：

初始状态下每个位置的父亲都是他本身， 当有一个数填到该位置时， 他的父亲就成了右边的一个数， 如果是末尾， 那父亲就是第一个数

set写法 ：

先将0 到 N - 1的数存进set， 每次lower_bound查找一下， 如果找到了就将该位置删掉， 否则把它放在开头位置， 再把开头位置删掉即可

Accepted code

并查集 ：

#include<bits/stdc++.h>
#include<unordered_map>
using namespace std;
 
#define sc scanf
#define ls rt << 1
#define rs ls | 1
#define Min(x, y) x = min(x, y)
#define Max(x, y) x = max(x, y)
#define ALL(x) (x).begin(),(x).end()
#define SZ(x) ((int)(x).size())
#define MEM(x, b) memset(x, b, sizeof(x))
#define lowbit(x) ((x) & (-x))
#define P2(x) ((x) * (x))
 
typedef long long ll;
const int MOD = 1e9 + 7;
const int MAXN = 1e6 + 10;
const int INF = 0x3f3f3f3f;
inline ll fpow(ll a, ll b){ ll r = 1, t = a; while (b){ if (b & 1)r = (r*t) % MOD; b >>= 1; t = (t*t) % MOD; }return r; }
 
ll p[MAXN], n, m;
ll s[MAXN];
int find_(int x) {
    while (x != p[x]) x = p[x] = p[p[x]];  // 查找父亲
    return x;
}
 
int main()
{
    cin >> n; 
    for (ll i = 0; i < n; i++) p[i] = i;
    for (ll i = 0; i < n; i++) {
        ll tmp, cnt, fa;
        sc("%lld", &tmp); 
        cnt = tmp % n;
        fa = find_(cnt); 
        s[fa] = tmp; // 存数
        if (fa == n - 1) p[fa] = find_(p[0]); // 最后一个数父亲改为第一个
        else p[fa] = find_(fa + 1); // 否则父亲就是后面一个
    }
    for (ll i = 0; i < n; i++) cout << s[i] << " ";
    cout << endl;
    return 0;
}

set写法

#include<bits/stdc++.h>
#include<unordered_map>
using namespace std;

#define sc scanf
#define ls rt << 1
#define rs ls | 1
#define Min(x, y) x = min(x, y)
#define Max(x, y) x = max(x, y)
#define ALL(x) (x).begin(),(x).end()
#define SZ(x) ((int)(x).size())
#define MEM(x, b) memset(x, b, sizeof(x))
#define lowbit(x) ((x) & (-x))
#define P2(x) ((x) * (x))

typedef long long ll;
const int MOD = 1e9 + 7;
const int MAXN = 1e6 + 10;
const int INF = 0x3f3f3f3f;
inline ll fpow(ll a, ll b){ ll r = 1, t = a; while (b){ if (b & 1)r = (r*t) % MOD; b >>= 1; t = (t*t) % MOD; }return r; }

set <int> st;
ll p[MAXN], n;
ll m;

int main()
{
	cin >> n;
	for (int i = 0; i < n; i++) st.insert(i); // 位置都存进去
	for (int i = 0; i < n; i++) {
		sc("%lld", &m);
		int tmp = m % n;
		if (!p[tmp]) { p[tmp] = m, st.erase(tmp); continue; }
		set <int> ::iterator it, item;
		it = st.lower_bound(tmp);
		if (it == st.end()) p[*st.begin()] = m, st.erase(st.begin()); //后面没有
		else p[*it] = m, st.erase(it);
	}
	for (int i = 0; i < n; i++) cout << p[i] << " ";
	cout << endl;
	return 0;
}