XDU-1153 万神的线段 （排序）

1153: 万神的线段

时间限制:  1 Sec   内存限制:  128 MB
http://acm.xidian.edu.cn/problem.php?id=1153
[ 提交 ][ 状态 ][ 讨论版 ]

题目描述

输入

输出

对于每组数据输出1行，包含1个整数，即平行或共线的线段对数。

样例输入

3
0 0 1 0
1 0 1 1
0 0 1 1
4
0 0 1 1
2 2 3 3
1 2 3 4
2 3 3 3
2
0 0 1 0
1 1 0 1

样例输出

0
3
1

提示

对于第二组样例，第 1 条线段与第 2 条共线，与第 3 条平行，第 2 条线段与第 3 条平行。

对于第三组样例，请注意线段是无向的，因此输入的两条线段平行。

读入线段数据后，直接按照“斜率”排序即可，相互平行的线段必定在一起，统计平行的线段为cnt，则这些平行的线段能构成cnt*(cnt-1)/2对平行线段

注意：cnt得用long long；特判斜率不存在的情况；

我将除法转化为乘法，比较斜率，没有精度误差，不过比赛时WA了好久，最后终于发现：dx可能为负值，所以在存储时，若dx为负值，则dx、dy均取相反数

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

struct Node {
    long long dx,dy;

    bool operator < (const Node& a) const {
        if(dx==0)
            return false;
        if(a.dx==0)
            return true;
        return dy*a.dx<a.dy*dx;
    }

    bool operator == (const Node& a) const {
        if(dx==0)
            return a.dx==0;
        if(a.dx==0)
            return false;
        return dy*a.dx==a.dy*dx;
    }
}p[100005];

int n;
long long x,y,xx,yy,cnt,ans;

int main() {
    while(1==scanf("%d",&n)) {
        for(int i=0;i<n;++i) {
            scanf("%lld%lld%lld%lld",&x,&y,&xx,&yy);
            p[i].dx=x-xx;
            p[i].dy=y-yy;
            if(p[i].dx<0) {
                p[i].dx=-p[i].dx;
                p[i].dy=-p[i].dy;
            }
        }
        sort(p,p+n);
        ans=0;
        cnt=1;
        for(int i=1;i<n;++i) {
            if(p[i]==p[i-1])
                ++cnt;//统计平行线段的个数
            else {
                ans+=cnt*(cnt-1)/2;//这些平行线段中互相匹配的结果
                cnt=1;
            }
        }
        if(p[n-1]==p[n-2])//若最后几个线段平行，则在循环中没有处理
            ans+=cnt*(cnt-1)/2;
        printf("%lld\n",ans);
    }
}