牛客挑战赛 34 D:拉普兰德的愿望(曼哈顿距离转切比雪夫距离)

在这里插入图片描述
直接求不好求,正难则反,用总答案 - 距离小于 d 的点对数

对于每一个点,扣除距离等于 d的那个边界,曼哈顿距离的边界是一个菱形不好处理
在这里插入图片描述
将曼哈顿距离转化为切比雪夫距离,边界是一个正方形。
在这里插入图片描述
做法是每个点按 x 排序,用树状数组维护 [x - d,x + d] 的所有 y 坐标,然后O(log) 统计即可。


代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define lowbit(i) (i & (-i))
const int maxn = 3e5 + 10;
const int p = 1e5;
int n,d,l;
struct node{
	int x,y;
}a[maxn];
ll sum[maxn];
bool cmp(node a,node b) {
	return a.x < b.x;
}
void upd(int p,int v) {
	for(int x = p; x <= 300000; x += lowbit(x))
		sum[x] += v;
}
ll qry(int p) {
	ll ans = 0;
	p = min(p,300000);
	if(p <= 0) return 0;
	for(int x = p; x; x -= lowbit(x))
		ans += sum[x];
	return ans;
}
int main() {
	scanf("%d%d%d",&n,&d,&l);
	for(int i = 1,x,y; i <= n; i++) {
		scanf("%d%d",&x,&y);
		a[i].x = x + y + p;
		a[i].y = x - y + p;
	}
	sort(a + 1,a + n + 1,cmp);
	upd(a[1].y,1);
	int l = 1,r = 2;
	ll ans = 0;
	for(int i = 1; i <= n; i++) {
		while(a[r].x < a[i].x + d && r <= n) {
			upd(a[r].y,1);
			r++;
		}
		while(a[l].x <= a[i].x - d && l <= n && l < r) {
			upd(a[l].y,-1);
			l++;
		}
		ans += qry(a[i].y + d - 1) - qry(a[i].y - d) - 1;
	}
	printf("%lld\n",1ll * n * (n - 1) / 2 - ans / 2);
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值