hdu 1558(线段相交+并查集）

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

思路：主要是判断线段相交，如果相交，就是要合并了，并且修改一下sum值就可以了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define MAXN 1111
#define eps 1e-7
int parent[MAXN];
int sum[MAXN];
int n;
struct Node{
    double x1,y1,x2,y2;
}node[MAXN];

void Initiate()
{
    for(int i=1;i<=n;i++){
        parent[i]=i;
        sum[i]=1;
    }
}

int Find(int x)
{
    if(x==parent[x])
        return x;
    parent[x]=Find(parent[x]);
    return parent[x];
}

void Union(int u,int v)
{
    int r1=Find(u),r2=Find(v);
    if(r1==r2)return ;
    parent[r1]=r2;
    sum[r2]+=sum[r1];
}

double Multiply1(const Node &p,const Node &q)
{
    return (p.x1-p.x2)*(q.y1-p.y1)-(p.y1-p.y2)*(q.x1-p.x1);
}

double Multiply2(const Node &p,const Node &q)
{
    return (p.x1-p.x2)*(q.y2-p.y1)-(p.y1-p.y2)*(q.x2-p.x1);
}

bool Judge(const Node &p,const Node &q)
{
    if(max(p.x1,p.x2)>=min(q.x1,q.x2)&&
    max(q.x1,q.x2)>=min(p.x1,p.x2)&&
    max(p.y1,p.y2)>=min(q.y1,q.y2)&&
    max(q.y1,q.y2)>=min(p.y1,p.y2)&&
    Multiply1(p,q)*Multiply2(p,q)<=eps&&
    Multiply1(q,p)*Multiply2(q,p)<=eps)
    return true;
    else return false;
}


int main()
{
    int _case,i,x,t=0;
    char str[2];
    scanf("%d",&_case);
    while(_case--)
    {
        if(t++)puts("");
        scanf("%d",&n);
        Initiate();
        i=1;
        while(n--){
            scanf("%s",str);
            if(str[0]=='P'){
                scanf("%lf%lf%lf%lf",&node[i].x1,&node[i].y1,&node[i].x2,&node[i].y2);
                for(int j=1;j<i;j++){
                    if(Judge(node[j],node[i])){
                        Union(j,i);
                    }
                }
                i++;
            }else {
                scanf("%d",&x);
                x=Find(x);
                printf("%d\n",sum[x]);
            }
        }
    }
    return 0;
}