牛客练习赛50 B tokitsukaze and Hash Table

B tokitsukaze and Hash Table

https://ac.nowcoder.com/acm/contest/1080/B

实际上我们就是要维护每个点右边（循环）第一个空位的位置，那么我们发现，插入一个新的节点可能导致一段区间右边（循环）的空位发生改变，不好直接暴力维护。我们模拟几下以后发现这个过程就类似于把几个区间拼接在一起，我们就考虑能否用并查集去维护这个拼接过程。

此时我们考虑：我们将右边（循环）第一个空位相同的的位置视为同一集合，我们假设这个空位是 $x$ ，那么我们就令这个集合的根为 $x$ （ $x$ 必然在集合内，因为其自己循环向右看第一个空位同样是 $x$ ）。我们定义某位置 $i$ 所属集合的根为 $ff(i)$ ，那么起初对于每个位置 $i$ ，必然有 $ff(i)=i$ 。那么倘若当前某元素想要插入位置 $idx$ ，那么其实际上就是要插入 $ff(idx)$ ，此时我们考虑哪些集合会发生改变，我们可以发现，任何一个集合只有根被插入的时候其集合内的后方空位才可能发生改变，否则根节点永远都会作为一个空节点卡在那里，所以这个时候只有 $i$ 所属的集合其后方空位发生了改变，那么其整个集合的后方空位就等于 $ff(idx)+1$ 这个节点的后方空位，那么我们将 $i$ 所属集合合并到 $ff(idx)+1$ 所属的集合即可。

#include <bits/stdc++.h>

using namespace std;

const int N = 1e6 + 10;

int n, a[N], fa[N];

int ff(int x) {
	return fa[x] == x ? x : fa[x] = ff(fa[x]);
}

int main() {
#ifndef ONLINE_JUDGE
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif
	scanf("%d", &n);
	for (int i = 0; i < n; ++i) fa[i] = i;
	for (int i = 0, x; i < n; ++i) {
		scanf("%d", &x);
		int idx = x % n;
		int nxt = ff(idx);
		a[nxt] = x;
		fa[nxt] = fa[ff((nxt + 1) % n)];
	}
	for (int i = 0; i < n; ++i) {
		printf("%d%c", a[i], " \n"[i == n - 1]);
	}
}

如果不嫌麻烦。。。也可以傻傻的像我一样维护这个序列的相邻联通块，也同样是可写的：

#include <bits/stdc++.h>

using namespace std;

const int N = 1e6 + 10;

int n, a[N];
int fa[N], sz[N], vis[N];

int ff(int x) {
	return fa[x] == x ? x : fa[x] = ff(fa[x]);
}

void merge(int x, int y) {
	int fx = ff(x), fy = ff(y);
	if (fx != fy) {
		fa[fy] = fx;
		sz[fx] += sz[fy];
	}
}

void act(int x) {
	vis[x] = 1;
	if (vis[(x - 1 + n) % n]) merge((x - 1 + n) % n, x);
	if (vis[(x + 1) % n]) merge(x, (x + 1) % n);
}


int main() {
#ifndef ONLINE_JUDGE
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif
	scanf("%d", &n);
	for (int i = 0; i < n; ++i) {
		fa[i] = i;
		sz[i] = 1;
	}
	for (int i = 0, x; i < n; ++i) {
		scanf("%d", &x);
		int idx = x % n;
		if (vis[idx]) {
			int fidx = ff(idx);
			idx = (fidx + sz[fidx]) % n; 
		}
		a[idx] = x;
		act(idx);
	}
	for (int i = 0; i < n; ++i) {
		printf("%d%c", a[i], i == n - 1 ? '\n' : ' ');
	}
}

如果像我一样吃饱了没事干，还可以写一发线段树上二分：

#include <bits/stdc++.h>
#define read(a) {char c;while((c=getchar())>47) a=a*10+(c^48);}
#define lson (rt<<1)
#define rson (rt<<1|1)

using namespace std;

const int N = 1e6 + 10;

int n, a[N], vis[N];
int tree[N << 2];

void update(int rt, int l, int r, int pos) {
	if (l == r) {
		tree[rt] = 1;
		return ;
	}
	int mid = (l + r) >> 1;
	if (pos <= mid) update(lson, l, mid, pos);
	else update(rson, mid + 1, r, pos);
	tree[rt] = tree[lson] + tree[rson];
}

int query(int rt, int l, int r, int L, int R) {
	if (L <= l && r <= R) {
		if (tree[rt] == r - l + 1) return -1;
		if (l == r) return l - 1;
		int mid = (l + r) >> 1;
		if (tree[lson] == mid - l + 1) return query(rson, mid + 1, r, L, R);
		return query(lson, l, mid, L, R);
	}
	else {
		int mid = (l + r) >> 1;
		if (mid >= L) {
			int ans = query(lson, l, mid, L, R);
			if (~ans) return ans;
		}
        if (mid < R) return query(rson, mid + 1, r, L, R);
        return -1;
	}
}

int main() {
#ifndef ONLINE_JUDGE
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif
	scanf("%d", &n);
	for (int i = 0; i < n; ++i) {
		int x; scanf("%d", &x);
		int idx = x % n;
		if (vis[idx]) {
			int idx1 = query(1, 1, n, idx + 1, n);
			int idx2 = query(1, 1, n, 1, idx + 1);
			idx = (idx1 == -1 ? idx2 : idx1);
		}
		a[idx] = x;
		vis[idx] = 1;
		update(1, 1, n, idx + 1);
	}
	for (int i = 0; i < n; ++i) {
		printf("%d%c", a[i], i == n - 1 ? '\n' : ' ');
	}
}