hdu 1558 Segment set 计算几何+并查集★

#include <cstdio>
#include <iostream>
#include <string.h>
using namespace std;

const double EPS = 1e-10;
#define MAX 1001

struct point    //点
{
    double x,y;
};

struct line     //线
{
    point a,b;
}l[MAX];

int father[MAX],num[MAX];
double Max(double a,double b)   {return a>b?a:b;}
double Min(double a,double b)   {return a>b?b:a;}

// 判断两线段是否相交（非规范相交）
bool inter(line l1,line l2)
{
    point p1,p2,p3,p4;
    p1=l1.a;p2=l1.b;
    p3=l2.a;p4=l2.b;

    if( Min(p1.x,p2.x)>Max(p3.x,p4.x) ||
       Min(p1.y,p2.y)>Max(p3.y,p4.y) ||
       Min(p3.x,p4.x)>Max(p1.x,p2.x) ||
       Min(p3.y,p4.y)>Max(p1.y,p2.y) )
        return 0;   //直接没有相交
    double k1,k2,k3,k4;
    k1 = (p2.x-p1.x)*(p3.y-p1.y) - (p2.y-p1.y)*(p3.x-p1.x);
    k2 = (p2.x-p1.x)*(p4.y-p1.y) - (p2.y-p1.y)*(p4.x-p1.x);
    k3 = (p4.x-p3.x)*(p1.y-p3.y) - (p4.y-p3.y)*(p1.x-p3.x);
    k4 = (p4.x-p3.x)*(p2.y-p3.y) - (p4.y-p3.y)*(p2.x-p3.x);
    return (k1*k2<=EPS && k3*k4<=EPS);
}

//初始化函数
void Init(int n)
{
    int i;
    for(i=1;i<=n;i++)
    {
        father[i]=i;
        num[i]=1;
    }
}

int Find(int x)
{
    while(father[x]!=x)
        x=father[x];
    return x;
}

void combine(int a,int b)
{
    int temp_a,temp_b;
    temp_a=Find(a);
    temp_b=Find(b);

    // 在合并集合的时候，相应集合所含的个数也要合并
    if(temp_a!=temp_b)
    {
        father[temp_a]=temp_b;
        num[temp_b]+=num[temp_a];
    }
}

int main()
{
    int test,i,n,k,js;
    char c;
    cin>>test;
    while(test--)
    {
        js=0;
        cin>>n;
        Init(n);
        while(n--)
        {
            cin>>c;
            // 判断是P还是Q
            if(c=='P')
            {
                ++js;
                cin>>l[js].a.x>>l[js].a.y>>l[js].b.x>>l[js].b.y;

                // 判断该线段与之前线段是否相交，相交则合并
                for(i=1;i<js;++i)
                {
                    if( inter(l[js],l[i]) )
                        combine(js,i);
                }
            }
            else
            {
                cin>>k;
                cout<<num[Find(k)]<<endl;
            }
        }
        // 格式！！！很重要，最后一组测试数据后无空行
        if(test)   cout<<endl;
    }
    return 0;
}