HDU5862(树状数组好题)

本文介绍了一种使用树状数组解决线段统计问题的方法。通过存储与坐标轴平行的线段信息,并利用树状数组高效统计相交线段的数量。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题意:给一些平行坐标轴的线段,。统计相交数

题解:可以这样考虑,把所有与x轴平行的线段存起来,用来枚举。把与y轴平行的线段的左右端点存起来。这样在枚举x线段时。就需要看x线段的左端点左边有多少,右端点有多少,相减就是中间重合的。如何维护这些数据呢?当然就得用树状数组了。,简单粗暴

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<algorithm>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<string>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f;
const ll INFF=0x3f3f3f3f3f3f3f3f;
const double pi=acos(-1.0);
const double eps=1e-9;
ll bit[200010];
struct node1
{
    int x1,x2,y;
    int id;
}node[200010];
int ax[200010];
bool cmp(node1 p1,node1 p2)
{
    if(p1.y!=p2.y)
        return p1.y<p2.y;
    return p1.id>p2.id;
}
ll sum(int i)
{
    ll ans=0;
    while(i)
    {
        ans+=bit[i];
        i-=(i&-i);
    }
    return ans;
}
void add(int i,int x,int len)
{
    while(i<=len)
    {
        bit[i]+=x;
        i+=i&(-i);
    }
}
int main()
{
    int t;scanf("%d",&t);
    while(t--)
    {
        int n;scanf("%d",&n);
        int cnt=0,tmp=0;
        for(int i=1;i<=n;i++)
        {
            int x1,x2,y1,y2;scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            if (x1 > x2) swap(x1, x2);
            if (y1 > y2) swap(y1, y2);
            if(x1==x2)
            {
                ax[cnt++]=x1;
                node[tmp].x1=x1;node[tmp].y=y1;node[tmp++].id=1;
                node[tmp].x1=x2;node[tmp].y=y2;node[tmp++].id=-1;
            }
            else
            {
                ax[cnt++]=x1;ax[cnt++]=x2;
                node[tmp].x1=x1;
                node[tmp].x2=x2;
                node[tmp].y=y2;
                node[tmp++].id=0;
            }
        }
        sort(ax,ax+cnt);
        int len=unique(ax,ax+cnt)-ax;
        sort(node,node+tmp,cmp);
        memset(bit,0,sizeof(bit));
        ll ans=0;
        for(int i=0;i<tmp;i++)
        {
            if(node[i].id==0)
            {
                int l=lower_bound(ax,ax+len,node[i].x1)-ax+1;
                int r=lower_bound(ax,ax+len,node[i].x2)-ax+1;
                ans+=sum(r)-sum(l-1);
            }
            else
            {
                int l=lower_bound(ax,ax+len,node[i].x1)-ax+1;
                add(l,node[i].id,len+1);
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值