ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

CF EDU 113 D - Inconvenient Pairs

2022-05-16 22:03:17  阅读:173  来源: 互联网

标签:Pairs int sum second CF add id Inconvenient xcnt


D - Inconvenient Pairs

思维

不方便的点对就是类似于,这种在同一行块或同一列块的两个点,他们的距离一定大于曼哈顿距离

其中红色为横向点对,紫色为纵向点对
image

所以可按 y 递增排序,找到每一个行块有多少个点,这一行块中的点对贡献为:\(\binom {cnt}2-\sum\binom {同一列的点的数量}2\)

注意当某个点的 y 就是行块分割的线时,它和任何点都不会构成 “横向” 的点对,所以可以一开始读入的时候记录下来,枚举到这个点就continue

求 “纵向” 点对数目同理

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include <cmath>
#include <map>

using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 3e5 + 10;
int n, m, k;
int row[N], col[N];
map<int, int> xcnt, ycnt;
map<int, bool> xban, yban;
struct Point
{
	int x, y;
}p[N];

bool cmp1(Point A, Point B)
{
	if (A.x == B.x)
		return A.y < B.y;
	return A.x < B.x;
}

bool cmp2(Point A, Point B)
{
	if (A.y == B.y)
		return A.x < B.x;
	return A.y < B.y;
}

void init()
{
	xcnt.clear();
	ycnt.clear();
	xban.clear();
	yban.clear();
}
int main()
{
	ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	int T;
	cin >> T;
	while(T--)
	{
		cin >> n >> m >> k;
		init();
		for (int i = 1; i <= n; i++)
		{
			cin >> col[i];
			xban[col[i]] = true;
		}
			
		for (int i = 1; i <= m; i++)
		{
			cin >> row[i];
			yban[row[i]] = true;
		}
			
		for (int i = 1; i <= k; i++)
			cin >> p[i].x >> p[i].y;
		sort(p + 1, p + k + 1, cmp2);
		int id = 2;
		ll ans = 0;
		for (int i = 1; i <= k; i++)
		{
			int x = p[i].x, y = p[i].y;
			if (yban[y])
				continue;
			if (y > row[id])
			{
				ll add = 0, sum = 0;
				for (auto t : xcnt)
				{
					sum += t.second;
					add -= t.second * (t.second - 1) / 2;
				}
				add += sum * (sum - 1) / 2;
				ans += add;
				xcnt.clear();
				while(y > row[id]) id++;
			}
			xcnt[x]++;
		}
		if (xcnt.size())
		{
			
			ll add = 0, sum = 0;
			for (auto t : xcnt)
			{
				sum += t.second;
				add -= t.second * (t.second - 1) / 2;
			}
			add += sum * (sum - 1) / 2;
			ans += add;
			xcnt.clear();
		}
		sort(p + 1, p + k + 1, cmp1);
		id = 2;
		for (int i = 1; i <= k; i++)
		{
			int x = p[i].x, y = p[i].y;
			if (xban[x])
				continue;
			if (x > col[id])
			{
				ll add = 0, sum = 0;
				for (auto t : ycnt)
				{
					sum += t.second;
					add -= t.second * (t.second - 1) / 2;
				}
				add += sum * (sum - 1) / 2;
				ans += add;
				ycnt.clear();
				while(x > col[id]) id++;
			}
			ycnt[y]++;
		}
		if (ycnt.size())
		{
			ll add = 0, sum = 0;
			for (auto t : ycnt)
			{
				sum += t.second;
				add -= t.second * (t.second - 1) / 2;
			}
			add += sum * (sum - 1) / 2;
			ans += add;
			ycnt.clear();
		}
		cout << ans << endl;
	}
	return 0;
}

标签:Pairs,int,sum,second,CF,add,id,Inconvenient,xcnt
来源: https://www.cnblogs.com/hzy717zsy/p/16278715.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有