CodeCraft-22 and Codeforces Round #795 (Div. 2) E. Number of Groups(贪心,区间问题)
链接
题意:给出一些区间,并且给出颜色,如果两个区间颜色不同并且有交集,则添加一条边,最后求有多少个连通分量。
思路:首先暴力贪心很容易想,对于每个区间,我们可以把他按照左端点排序,左端点代表进入区间,右端点代表删除区间,然后把他加入对应颜色的set,当set里面有交点就进行一次并查集维护连通块,但是这样做复杂度仍会下降到
O
(
n
2
)
O(n^2)
O(n2),我们可以发现,因为我们是按左端点排序,所以每次联通了区间只后,实际上只需要保留右端点最大的那个区间就行了,因为我们统计的只是连通块,不是具体的形态,而最大的那个可以覆盖较小的对应的交集,所以只需要保留最大的那个,执行之前先把不合法的情况弹出。
#include<bits/stdc++.h>
using namespace std;
struct node {int c, l, r, id;}a[100010];
struct Node {int x, c, id;};
bool cmp(Node a, Node b) {if(a.x == b.x) return a.c < b.c; return a.x < b.x;}
const int N = 1e5+10;
int fa[N], siz[N];
int find(int n) {return n == fa[n] ? n : fa[n] = find(fa[n]);}
void solve()
{
int n;
cin >> n;
vector<Node> vec;
for (int i = 1; i <= n; i++) fa[i] = i;
for (int i = 1; i <= n; i++) {
cin >> a[i].c >> a[i].l >> a[i].r;
a[i].id = i;
vec.push_back({a[i].l, -1, i});
vec.push_back({a[i].r, 1, i});
}
sort(vec.begin(), vec.end(), cmp);
set<pair<int, int> >st[2];
int ans = 0;
for (int i = 0; i < vec.size(); i++) {
int id = vec[i].id;
if (vec[i].c == -1) {
st[a[id].c].insert({a[id].r, id});
int c = a[id].c ^ 1;
while (!st[c].empty() && st[c].begin()->first < vec[i].x) st[c].erase(st[c].begin());
int f = 0;
while (!f && !st[c].empty()) {
int x = st[c].begin()->second, y = id;
if (st[c].size() > 1) st[c].erase(st[c].begin());
else f = 1;
x = find(x), y = find(y);
if (x == y) continue;
fa[x] = y;
}
}else {st[a[id].c].erase({a[id].r, id});}
}
for (int i = 1; i <= n; i++) ans += (find(i) == i);
cout << ans << endl;
}
int main()
{
int T;
cin >> T;
while (T--) {solve();}
return 0;
}