HDU5862（树状数组好题）

最新推荐文章于 2021-08-16 20:21:11 发布

Phoenix丶HN

最新推荐文章于 2021-08-16 20:21:11 发布

阅读量422

点赞数 2

CC 4.0 BY-SA版权

分类专栏：线段树数据结构

本文链接：https://blog.youkuaiyun.com/nucshiyilang/article/details/75675111

数据结构同时被 2 个专栏收录

27 篇文章

订阅专栏

线段树

20 篇文章

订阅专栏

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

摘要生成于 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;
}