Educational Codeforces Round 68 E. Count The Rectangles
传送门
题意：
给出不超过\(n,n\leq 5000\)条直线，问共形成多少个矩形。
思路：
考虑到\(n\)的范围不大，所以可以暴力枚举两条平行的直线，接下来处理的就是与其垂直的直线的数量。
满足相交成矩形有两个条件，假如我们枚举的直线是垂直于\(x\)轴的，那么两个条件即为\(low\leq y_i\leq high,x_{i,0}\leq left,right\leq x_{i,1}\)。
所以我们可以考虑扫描线的思想，根据左右端点选择插入/删除直线，插入的时候利用权值树状数组维护其\(y_i\)值。之后对于竖直的两条直线，查询范围在\(low\)~\(high\)之间的水平直线个数即可。设其为\(x\)，那么对答案的贡献就为\(C_{x}^2\)。
注意一下\(low\)可能大于\(high\)的情况。
代码如下：
#include
using namespace std;
typedef long long ll;
const int N = 50005;
int n;
int x[N][2], y[N][2];
int X[N], Y[N];
struct seg1{
int x, y1, y2;
bool operator < (const seg1 &A) const {
return x < A.x;
}
}s1[N];
struct seg2{
int y, x1, x2;
}s2[N];
vector v1[N], v2[N], v3[N];
int c[N];
int lowbit(int x) {
return x & (-x);
}
void add(int x, int v) {
for(int i = x; i < N; i += lowbit(i)) c[i] += v;
}
int query(int p) {
int ans = 0;
for(int i = p; i; i -= lowbit(i)) ans += c[i];
return ans;
}
int main() {
ios::sync_with_stdio(false); cin.tie(0);
cin >> n;
int D1 = 0, D2 = 0;
for(int i = 1; i <= n; i++) {
cin >> x[i][0] >> y[i][0] >> x[i][1] >> y[i][1];
X[++D1] = x[i][0]; X[++D1] = x[i][1];
Y[++D2] = y[i][0]; Y[++D2] = y[i][1];
}
sort(X + 1, X + D1 + 1); sort(Y + 1, Y + D2 + 1);
D1 = unique(X + 1, X + D1 + 1) - X - 1;
D2 = unique(Y + 1, Y + D2 + 1) - Y - 1;
int t1 = 0, t2 = 0;
for(int i = 1; i <= n; i++) {
x[i][0] = lower_bound(X + 1, X + D1 + 1, x[i][0]) - X;
x[i][1] = lower_bound(X + 1, X + D1 + 1, x[i][1]) - X;
y[i][0] = lower_bound(Y + 1, Y + D2 + 1, y[i][0]) - Y;
y[i][1] = lower_bound(Y + 1, Y + D2 + 1, y[i][1]) - Y;
if(x[i][0] > x[i][1]) swap(x[i][0], x[i][1]);
if(y[i][0] > y[i][1]) swap(y[i][0], y[i][1]);
if(x[i][0] == x[i][1]) s1[++t1] = seg1{x[i][0], y[i][0], y[i][1]};
if(y[i][0] == y[i][1]) s2[++t2] = seg2{y[i][0], x[i][0], x[i][1]};
if(y[i][0] == y[i][1]) v1[x[i][0]].push_back(t2);
}
sort(s1 + 1, s1 + t1 + 1);
for(int i = 1; i <= t1; i++) v3[s1[i].x].push_back(i);
ll res = 0;
int bound = D1;
for(int x = 1; x <= bound; x++) {
for(auto it : v2[x]) {
add(s2[it].y, -1);
}
for(auto it : v1[x]) {
add(s2[it].y, 1);
v2[s2[it].x2].push_back(it);
}
for(auto i : v3[x]) {
for(int j = x + 1; j <= bound; j++) {
for(auto k : v3[j]) {
int R = min(s1[i].y2, s1[k].y2), L = max(s1[i].y1, s1[k].y1);
if(L > R) continue;
int tmp = query(R) - query(L - 1);
res += 1ll * tmp * (tmp - 1) / 2;
}
for(auto it : v2[j]) add(s2[it].y, -1);
}
for(int j = x + 1; j <= bound; j++) {
for(auto it : v2[j]) add(s2[it].y, 1);
}
}
}
cout << res;
return 0;
}