POJ-3109-Inner Vertices

这个题参考了下http://www.hankcs.com/program/algorithm/poj-3109-inner-vertices.html,方法也差不多。发现扫描线还有这样的用法~

代码:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int inf=1<<29;
const int maxn=2e5+100;
struct Point
{
    int x;
    int y;
    bool operator < (const Point &a)const
    {
        if(y==a.y)
            return x<a.x;
        return y<a.y;
    }
}a[maxn];
int n,cnt,p[maxn],t[maxn];
bool isr[maxn],isl[maxn],vis[maxn];
int lowbit(int x)
{
    return x&(-x);
}
void Update(int x,int val)
{
    for(;x<maxn;t[x]+=val,x+=lowbit(x));
}
int Sum(int x)
{
    long long ans=0;
    for(;x>0;ans+=t[x],x-=lowbit(x));
    return ans;
}
void Init()
{
    memset(t,0,sizeof(t));
    sort(p,p+cnt);
    cnt=unique(p,p+cnt)-p;
    for(int i=0;i<n;i++)
    {
        a[i].x=lower_bound(p,p+cnt,a[i].x)-p+1;
        a[i].y=lower_bound(p,p+cnt,a[i].y)-p+1;
    }
    sort(a,a+n);
    memset(vis,0,sizeof(vis));
    for(int i=0;i<n;i++)
    {
        if(!vis[a[i].x])
        {
            isl[i]=1;
            vis[a[i].x]=1;
        }
        else
            isl[i]=0;
    }
    memset(vis,0,sizeof(vis));
    for(int i=n-1;i>=0;i--)
    {
        if(!vis[a[i].x])
        {
            isr[i]=1;
            vis[a[i].x]=1;
        }
        else
            isr[i]=0;
    }
}
void solve()
{
    long long ans=0;
    for(int i=0;i<n;)
    {
        int j=i;
        while(j<n-1&&a[j+1].y==a[j].y)
            j++;
        for(int k=i;k<=j;k++)
            if(!isl[k]&&isr[k])
                Update(a[k].x,-1);
        if(i!=j)
            ans+=Sum(a[j].x-1)-Sum(a[i].x);
        if(i<j)
            for(int k=i+1;k<j;k++)
                if(!isl[k]&&!isr[k])
                    ans--;
        while(i<=j)
        {
            if(isl[i]&&!isr[i])
                Update(a[i].x,1);
            i++;
        }
    }
    printf("%I64d\n",ans+n);
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        cnt=0;
        for(int i=0;i<n;i++)
        {
            scanf("%d%d",&a[i].x,&a[i].y);
            p[cnt++]=a[i].x;
            p[cnt++]=a[i].y;
        }
        Init();
        solve();
    }
    return 0;
}


你可能感兴趣的:(树状数组,扫描线)