题目大意:有一个10001000 的矩阵,上面有很多个物品,当两个物品在同一条对角线上时,能相互攻击,给出n个物品的坐标,问能相互攻击的对数有多少。
大体思路:直接进行暴力循环求解会超时
1 如果两个元素在同一条平行于主对角线的对角线上,那么他们的行和列的差值相等,
2 如果两个元素在同一条平行于副对角线的对角线上,那么他们的行和列的和相等
然后就是分别统计,并且找出所有在同一对角线上元素个数,每种选两个 举例可发现n种元素任意两种组合可形成的对数有n(n-1)/2。
我学习到的小技巧:本题中lower_bound()函数和upper_bound()函数的妙用。**
#include<cstdio>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
int a[200005],b[200005];
int n;
long long cmp(int ant[])
{
long long sum=0;
int i=0;
while(i<n)
{
int p=lower_bound(ant,ant+n,ant[i])-ant;
int q=upper_bound(ant,ant+n,ant[i])-ant;
sum+=(q-p)*(q-p-1)/2;
i=q;
}
return sum;
}
int main()
{
scanf("%d",&n);
for(int i=0;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
a[i]=x-y;
b[i]=x+y;
}
sort(a,a+n);
sort(b,b+n);
long long ans=cmp(a)+cmp(b);
printf("%lld\n",ans);
return 0;
}