【codeforces 650A】【思维】【容斥】【数学】【求n个点中两种距离相等的点对数】

lzk_1049668876

于 2019-01-19 09:36:07 发布

阅读量222

点赞数 1

CC 4.0 BY-SA版权

分类专栏：思维容斥数学文章标签：思维容斥数学

本文链接：https://blog.youkuaiyun.com/running_acmer/article/details/86548896

思维同时被 3 个专栏收录

26 篇文章

订阅专栏

数学

4 篇文章

订阅专栏

容斥

1 篇文章

订阅专栏

本文提供了一道Codeforces平台上的题目650A的详细解题思路与代码实现。该题旨在求解给定点集中两点间距离相等的点对数量，通过坐标排序和容斥原理来优化计算过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【链接】https://codeforces.com/problemset/problem/650/A

【一句话题解】求n个点中两种距离相等的点对数。SX U SY =SX+SY +SX ∩ SY,x,y分别排序，map记录点对，容斥算重

【代码】

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
const int maxn = 1e6 + 6;
int x[maxn], y[maxn];

int cmp(int a, int b) {
	return a < b;
}
map<int, map<int,int> >mp,vis;

int main() {
	int n;
	while (~scanf("%d", &n)) {
		for (int i = 0; i < n; i++) {
			scanf("%d%d", &x[i], &y[i]);
			mp[x[i]][y[i]]++;
		}
		ll ans = 0;

		for (int i = 0; i < n; i++) {
			if (mp[x[i]][y[i]]>1&&!vis[x[i]][y[i]])ans-= (mp[x[i]][y[i]]-1LL)*(mp[x[i]][y[i]])/2;
			vis[x[i]][y[i]] = 1;
		}
		sort(x, x + n, cmp);
		sort(y, y + n, cmp);
		
		for (ll i = 0; i < n; i++) {
			if (i!=0&&x[i] == x[i - 1])continue;
			ll j = lower_bound(x, x + n, x[i] + 1)-x;
			ans += (j - i - 1LL)*(j - i )/2;
		}
		for (ll i = 0; i < n; i++) {
			if (i!=0&&y[i] == y[i - 1])continue;
			ll j = lower_bound(y, y + n, y[i] + 1) - y;
			ans += (j - i - 1LL)*(j - i )/2;
		}
		cout << ans << endl;
	}
}

/*
3
1 1
7 5
1 5

*/

/*

6
0 0
0 1
0 2
-1 1
0 1
1 1

*/

/*

3
1 2
1 2
1 2

*/