hdu 1558 (线段相交+并查集) Segment set

题目:http://acm.hdu.edu.cn/showproblem.php?pid=1558

题意是在坐标系中,当输入P(注意是大写,我当开始就wa成了小写)的时候输入一条线段的起点坐标和终点坐标,当输入Q的时候输入n,然后输出与第n条线段相交的线段有多少条

首先判断线段是否相交,在算法导论p577上有介绍

线段A(x1,y1)-B(x2,y2),所在直线L1方程为F1(x,y)=0;
线段C(x3,y3)-D(x4,y4),所在直线L2方程为F2(x,y)=0;

如何判断两条线段有交点：(A,B在直线L2两侧) AND (C,D在直线L1两侧)。

用数学表达式来表示可以这样来表示：F2(x1,y1)*F2(x2,y2) >= 0 AND F1(x3,y3)*F1(x4,y4)>= 0; 等于0表示恰好在直线上

 

然后就是基础的并查集判断是否在一个集合以及集合内的线段的个数

#include<cstdio>
using namespace std;
int father[1001],num[1001];
void give()
{
    for (int i=1;i<=1000;i++)
    {
        father[i]=i;
        num[i]=1;
    }
}
int _find(int x)
{
    if (father[x]==x) return father[x];
    father[x]=_find(father[x]);
    return father[x];
}
double x1[1001],x2[1001],y1[1001],y2[1001];
double a[1001],b[1001];
int main()
{
    int t,n,sx,sy,k,i,q;
    char op;
    scanf("%d",&t);
        while (t--)
        {
            scanf("%d",&n);
            give();k=1;
            while (n--)
            {
                scanf(" %c",&op);
                if (op=='P')
                {
                    scanf("%lf %lf %lf %lf",&x1[k],&y1[k],&x2[k],&y2[k]);
                    a[k]=(y2[k]-y1[k])/(x2[k]-x1[k]);
                    b[k]=(x2[k]*y1[k]-x1[k]*y2[k])/(x2[k]-x1[k]);
                    for (i=1;i<k;i++)
                    {
                        int t1=0,t2=0;
                        if (!((a[k]*x1[i]-y1[i]+b[k])*(a[k]*x2[i]-y2[i]+b[k])>0))
                            t1=1;
                        if (!((a[i]*x1[k]-y1[k]+b[i])*(a[i]*x2[k]-y2[k]+b[i])>0))
                            t2=1;
                        if (t1==1&&t2==1)
                        {
                            sx=_find(k);
                            sy=_find(i);
                            if (sx!=sy){
                            father[sx]=sy;
                            num[sy]+=num[sx];
                            }
                        }
                    }
                    k++;
                }
                else
                {
                    scanf("%d",&q);
                    printf("%d\n",num[_find(q)]);
                }
            }
            if (t) printf("\n");
        }
    return 0;
}