JNUOJ 1187

Time Limit: 10000ms

Memory Limit: 262154KB

64-bit integer IO format: %lld      Java class name: Main

 

木衫同学和凌小小都是军队的指挥官，他们共同管理着一队哨兵，每个哨兵都有一个位置(x,y)。但是木衫同学和凌小小计算距离的方法不一样，木衫同学计算第i个人和第j个人的距离利用以下公式 |xi - xj| + |yi - yj|，而凌小小计算时利用以下公式sqrt( (xi-xj)^2+(yi-yj)^2 )。现在问有多少组哨兵，他们之间的距离运用两种公式计算出来的结果是一样的。

Input

第一行输入一个整数T，代表有T组数据
每组数据第一行输入n（1 ≤ n ≤ 200000），代表哨兵的个数。
下面的n行，每行有两个整数，代表当前哨兵的坐标，(|xi|, |yi| ≤ 10^9)。

Output

每组数据输出一行，这一行只有一个整数，即符合条件的哨兵的组数。

Sample Input

2
3
1 1
7 5
1 5
6
0 0
0 1
0 2
-1 1
0 1
1 1

Sample Output

2
11

单纯暴力当然是要超时的……其实满足|xi - xj| + |yi - yj| = sqrt( (xi-xj)^2+(yi-yj)^2 )的两个点，只能是在同一条水平线或者同一条竖直线上

所以可以把不用计算两个距离，而是通过判断是否在同一条水平线或者同一条竖直线上来判断是否满足|xi - xj| + |yi - yj| = sqrt( (xi-xj)^2+(yi-yj)^2 )

#include<cstdio>
#include<cmath>
#include<cstdlib>
long long ans;
int n;
struct type{
    double x,y;
}p[200000+5];
bool check(type a,type b)
{
    if(a.x == b.x || a.y == b.y) return true;
    return false;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++) scanf("%lf%lf",&p[i].x,&p[i].y);
        ans=0;
        for(int i=1;i<n;i++)
        {
            for(int j=i+1;j<=n;j++)
            {
                if(check(p[i],p[j])) ans++;
            }
        }
        printf("%lld
",ans);
    }
}

这大概要耗时4秒左右。

所以可以考虑把用map把出现过的水平线（y=k的k作为key值）或者竖直线记录下来（x=k的k作为key值）,然后把出现过的两个哨兵在相同的点上的情况也记录下来，

这样，每输入一个哨兵的坐标，ans+= “与他在同一条水平线上的人” + “与他在同一条竖直线上的人” - “与他在同一个坐标上的人（因为这样的人重复算了两次）”

#include<cstdio>
#include<map>
using namespace std;
typedef long long ll;
pair<ll,ll> now;
map<ll,ll> a,b;
map<pair<ll,ll>,ll> same;
int n;
ll ans,x,y;
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        a.clear();b.clear();same.clear();
        ans=0;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&x,&y);
            ans=ans+a[x]+b[y]-same[make_pair(x,y)];
            a[x]++;
            b[y]++;
            same[make_pair(x,y)]++;
        }
        printf("%d
",ans);
    }
}

 这样耗时大概是1秒左右。