题目描述
有一个圆,圆周上按顺时针方向给出 2n 个点。第 i 个点的颜色是
colori ,其中数据保证 1≤colori≤n ,而且每种不同的颜色有且只有两个点。这意味着有两个颜色是1的点,有两个颜色是2的点,有两个颜色是3的点,….有两个颜色是 n 的点。不存在位置重叠的点。颜色相同的两个点之间连一条边(线段)。现在的问题是:有多少对边是交叉的?
输入格式 1699.in
第一行,一个整数
n 。 1≤n≤50001 。
第二行, 2n 个整数,第 i 个整数是表示第i 个点的颜色 colori 。
输出格式 1699.out
一个整数。
输入样例 1699.in
4
3 2 4 4 1 3 2 1
输出样例 1699.out
3
样例解释
边1与边2交叉;
边1与边3交叉;
边2与边3交叉。
初看题目,容易想到
O(n2)
的做法:记录每条边的两个端点位置
xi
和
yi
,枚举任意两条边
i
和
如果我们已经确定了一条边,那么显然与其产生交叉的边的数量就是在当前边的两个端点之间出现的不成对的端点的个数。道理很浅显:如果在当前边两个端点之间包含的是另一条完整的边(如样例中3和4),那么不产生交叉。如果两条边之间互不交叉(如样例中4和1),那么端点之间的位置不存在包含关系。换而言之,对于边
i
和边
与之前的做法类似,我们先通过预处理方便地求得
xi
。然后,从第一个点开始顺时针遍历接下来的每一个点
j
。每次遍历,对“当前点的颜色被遍历”这个状态,用数组计数做一个标记(vis[color[j]]=true
)。当某次遍历遇到一个点,该点的颜色已被遍历过时(vis[color[j]]==true
),意味着访问完了一条完整的边。这时要统计的是
如何统计呢?先放下这个,不妨将问题稍加转化。最初时给每个点一个权值1,每处理完一条边之后将其两个端点的权值改为0(单点修改)。而由于我们的遍历是按顺时针顺序进行的,对边的处理也就是按其结束端点的先后顺序执行。因此在不产生交叉的两种情况中,另一边要么是结束端点在当前边的结束端点之前,已经先于当前边处理完了(即样例中3和4的情况);要么是另一边的开始端点在当前边的结束端点之后,没有重复。(即样例中4和1的情况)。
也就是说,对于当前边,所求的即为在其两个端点之间权值为1的点的个数(区间查询)。最终简化问题,我们要维护的其实就是一个前缀和。可以用树状数组或线段树实现。
参考代码:
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
using namespace std;
#define lowbit(x) (x & -x)
const int maxn = 5e4 + 10;
int n;
int f[maxn << 1], t[maxn << 1];
//树状数组操作
int query(int k) {
int sum = 0;
while (k) { sum += t[k]; k -= lowbit(k); }
return sum;
}
void update(int k, int v) {
while (k <= n << 1) { t[k] += v; k += lowbit(k); }
}
int main(void) {
freopen("1699.in", "r", stdin);
freopen("1699.out", "w", stdout);
scanf("%d", &n);
int ans = 0;
for (int i = 1; i <= n << 1; i++) {
int m;
scanf("%d", &m);
if (f[m]) { //如果当前端点是某条边的结束端点
ans += query(i - 1) - query(f[m]); //统计该边范围内1的个数
update(f[m], -1); //将起始点的权值加上-1(即修改成了1)
} else { f[m] = i; update(i, 1); } //否则当前是开始点,相应处理
}
printf("%d\n", ans);
return 0;
}