
Part of series: PS
16566 BOJ
문제
접근
분리 집합을 이용해 해결하였다. 자기 자신을 포함해 남아 있는 카드 중 자신보다 크거나 같은 카드를 parent로 설정하였다. 이러한 분리 집합 외에도 정렬된 배열을 set으로 만들어 해결하거나 세그먼트 트리를 이용해 해결할 수 있다.
코드
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ull, ull> pull;
typedef const ll cll;
#define FOR(a, A) for (ll a = 0; a < A; ++a)
cll N = 4e6, M = 4e6, K = 1e4;
ll n, m, k, cards[M] = {}, parents[N + 2] = {};
int main(void) {
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
cin >> n >> m >> k;
FOR(i, m) { cin >> cards[i]; }
sort(cards, cards + n);
set<ll> s(cards, cards + n);
FOR(i, k) {
ll num;
cin >> num;
cout << (num = *s.upper_bound(num)) << "\n";
s.extract(num);
}
return 0;
}
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef const ll cll;
#define FOR(a, A) for (ll a = 0; a < A; ++a)
cll N = 4e6, M = 4e6, K = 1e4;
ll n, m, k, cards[M] = {}, parents[N + 2] = {};
ll ascend(ll node) {
if (node == parents[node]) {
return node;
}
return parents[node] = ascend(parents[node]);
}
inline void merge(ll node, ll to) { parents[ascend(node)] = to; }
int main(void) {
ios::sync_with_stdio(false);
cin.tie(NULL);
cout.tie(NULL);
cin >> n >> m >> k;
FOR(i, m) {
cin >> cards[i];
parents[cards[i]] = cards[i];
}
for (ll num = n + 1, parent; num >= 0; --num) {
if (parents[num] == num) {
parent = num;
} else {
parents[num] = parent;
}
}
FOR(i, k) {
ll num;
cin >> num;
cout << (num = ascend(num + 1)) << "\n";
merge(num, num + 1);
}
return 0;
}