题干:
链接: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插入。
解题报告:
这题做法不少,可以直接set维护第一个空位,二分查找即可。
还可以用并查集维护空位,每次在x位置插入数后,合并x和(x+1)%n,也不难写。
AC代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<map>
#include<vector>
#include<set>
#include<string>
#include<cmath>
#include<cstring>
#define FF first
#define SS second
#define ll long long
#define pb push_back
#define pm make_pair
using namespace std;
typedef pair<int,int> PII;
const int MAX = 2e6 + 5;
set<int> ss;
int ans[MAX];
int main()
{int n;cin>>n;for(int i = 0; i<n; i++) ss.insert(i);for(int x,i = 1; i<=n; i++) {scanf("%d",&x);auto it = ss.lower_bound(x%n );if(it == ss.end()) ans[*ss.begin()] = x,ss.erase(ss.begin());else ans[*it] = x,ss.erase(it);}for(int i = 0; i<n; i++) printf("%d%c",ans[i],i == n-1 ? '\n' :' ');return 0 ;
}
附一个并查集代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<map>
#include<vector>
#include<set>
#include<string>
#include<cmath>
#include<cstring>
#define FF first
#define SS second
#define ll long long
#define pb push_back
#define pm make_pair
using namespace std;
typedef pair<int,int> PII;
const int MAX = 2e6 + 5;
int f[MAX];
int getf(int v) {return f[v] == v ? v : f[v] = getf(f[v]);
}
void merge(int u,int v) {int t1 = getf(u),t2 = getf(v);f[t1] = t2;
}
int ans[MAX];
int main()
{int n;cin>>n;for(int i = 0; i<=n; i++) f[i] = i;for(int x,i = 1; i<=n; i++) {scanf("%d",&x);int pos = x%n;int fa = getf(pos);ans[fa] = x;merge(fa,(fa+1)%n);}for(int i = 0; i<n; i++) printf("%d%c",ans[i],i == n-1 ? '\n' : ' ');return 0 ;
}