这道题有个很简单的思路,也有一个很难的思路,这个很难的思路用到了树状数组(但是是大佬写的),而简单的思路仅仅用到了归并排序求逆序对(也是一个大佬写的),而我连简单的思路都没想到,(*/ω\*)。
先说说简单的思路吧。
简单思路
cpp
using i64 = long long;
i64 ans;
void mergesort(i64 left, i64 right, std::vector<i64>& a, std::vector<i64>& b) {
if (left < right) {
i64 mid = (left + right) / 2;
mergesort(left, mid, a, b);
mergesort(mid + 1, right, a, b);
i64 i = left, j = mid + 1, ie = mid, je = right,k=0;
while (i <= ie && j <= je) {
if (a[i] <= a[j])b[k++] = a[i++];
else {
b[k++] = a[j++];
//求逆序对的核心代码
ans += ie - i + 1;
}
}
while (i <= ie)b[k++] = a[i++];
while (j <= je)b[k++] = a[j++];
for (i64 h = left; h <= right; h++)a[h] = b[h - left];
}
}
void solve() {
int n;
ans = 0;
std::cin >> n;
std::vector<i64>a(n), b(n),c(n),d(n),e(n);
std::iota(d.begin(), d.end(), 0);
for (int i = 0; i < n; i++)std::cin >> a[i] >> b[i];
std::sort(d.begin(), d.end(),
[&](int i, int j) {
return a[i] < a[j];
});
for (int i = 0; i < n; i++)c[i] = b[d[i]];
mergesort(0, n - 1, c,e);
std::cout << ans << '\n';
}
int main() {
int t;
std::cin >> t;
while (t--) {
solve();
}
return 0;
}
只需要把起点从小到大排,终点也要跟着动,然后求终点的逆序对即可,我们要求的是一个区间全包含了多少个区间,在外面按照起点从小到大排序的时候,我们已经确保的下一个起点一定是包含于上一个起点的了,那么我们只要找到它对应的终点的左边的终点有多少个比它小的,及全包含的区间有多少个,说白了就是总的逆序对个数即可。
困难思路
这个代码是jiangly大佬写的,也许人家认为比逆序对的思路简单吧,(*/ω\*),我太菜了。
树状数组我也没完全理解,但是不妨碍我使用它
cpp
using i64 = long long;
template <typename T>
struct Fenwick {
int n;
//a里面存的是一个区间,由大区间到小区间划分,起点都为0
std::vector<T> a;
//构造函数,不传参默认为0
Fenwick(int n_ = 0) {
init(n_);
}
//初始化n和数组a,assign(大小,类型{});
void init(int n_) {
n = n_;
a.assign(n, T{});
}
//右端点和权值
void add(int x, const T& v) {
//更新父节点,
for (int i = x + 1; i <= n; i += i & -i) {
a[i - 1] = a[i - 1] + v;
}
}
T sum(int x) {
T ans{};
for (int i = x; i > 0; i -= i & -i) {
ans = ans + a[i - 1];
}
return ans;
}
T rangeSum(int l, int r) {
return sum(r) - sum(l);
}
int select(const T& k) {
int x = 0;
T cur{};
for (int i = 1 << std::log2(n); i; i /= 2) {
if (x + i <= n && cur + a[x + i - 1] <= k) {
x += i;
cur = cur + a[x - 1];
}
}
return x;
}
};
void solve() {
int n;
std::cin >> n;
//起点,终点
std::vector<int> a(n), b(n), v;
//预留两倍n空间
v.reserve(2 * n);
for (int i = 0; i < n; i++) {
std::cin >> a[i] >> b[i];
v.push_back(a[i]);
v.push_back(b[i]);
}
std::sort(v.begin(), v.end());
//相当于建了个系,因为每个起点或者终点都是不同的,把它放在坐标系(v)对应的位置
//由于其唯一性,所以使用二分查找必定找到的是对应值,
for (int i = 0; i < n; i++) {
//找到a[i]和b[i]混合元素中第一个>=a[i]的值的下标
a[i] = std::lower_bound(v.begin(), v.end(), a[i]) - v.begin();
//找到a[i]和b[i]混合元素中第一个>=b[i]的值的下标
b[i] = std::lower_bound(v.begin(), v.end(), b[i]) - v.begin();
}
//把起点和终点重新关联起来,且起点是有序的,有序起点排名对应无序终点排名
std::vector<int> f(2 * n, -1);
for (int i = 0; i < n; i++) {
f[a[i]] = b[i];
}
//创建了一个树状数组
Fenwick<int> fen(2 * n);
i64 ans = 0;
//总共只有2*n个点,i代表的是左端点,如果你的左端点比别人的左端点大,
//那么只有别人和你greeting的份。
for (int i = 2 * n - 1; i >= 0; i--) {
//当右端点不为空时,说明是一个有效线段,由于i的递减性,所以下一个i肯定在上一个i左边
if (f[i] != -1) {
ans += fen.sum(f[i]);
//传入右端点和权值
fen.add(f[i], 1);
}
}
std::cout << ans << "\n";
}
int main() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
int t;
std::cin >> t;
while (t--) {
solve();
}
return 0;
}
题解都写在注释里面了,树状数组不会并不妨碍我们使用,我们排序完后f中存储是有序的a[i]对应的b[i].由于是从起点大的一端枚举,所以会累加到小的一端且终点大于等于该值的线段上。最好先稍微了解一下树状数组的原理。就我目前的水平如果再碰到类似的题我应该会选择简单思路来写。