2020牛客暑假多校训练营 第二场 H Happy Triangle set 线段树 分类讨论

LINK:Happy Triangle

这道题很容易。

容易想到 a+b<x a<x<b x<a<b 其中等于的情况在第一个和第三个之中判一下即可。

前面两个容易想到set来找前驱后继 第三个 考虑比x大的数字中两个连着的数字差的最小值 直接用线段树维护一下即可。

code
//#include<bitsstdc++.h>
#include<iostream>
#include<iomanip>
#include<cstdio>
#include<cstring>
#include<string>
#include<ctime>
#include<cmath>
#include<cctype>
#include<cstdlib>
#include<queue>
#include<deque>
#include<stack>
#include<vector>
#include<algorithm>
#include<utility>
#include<bitset>
#include<set>
#include<map>
#define ll long long
#define db double
#define INF 1000000000000000ll
#define ldb long double
#define pb push_back
#define put_(x) printf("%d ",x);
#define get(x) x=read()
#define gt(x) scanf("%d",&x)
#define gi(x) scanf("%lf",&x)
#define put(x) printf("%d
",x)
#define putl(x) printf("%lld
",x)
#define gc(a) scanf("%s",a+1)
#define rep(p,n,i) for(RE int i=p;i<=n;++i)
#define go(x) for(int i=lin[x],tn=ver[i];i;tn=ver[i=nex[i]])
#define fep(n,p,i) for(RE int i=n;i>=p;--i)
#define vep(p,n,i) for(RE int i=p;i<n;++i)
#define pii pair<int,int>
#define mk make_pair
#define RE register
#define P 1000000007
#define gf(x) scanf("%lf",&x)
#define pf(x) ((x)*(x))
#define uint unsigned long long
#define ui unsigned
#define EPS 1e-4
#define sq sqrt
#define S second
#define F first
#define maxx 1000000000
#define l(p) t[p].l
#define r(p) t[p].r
#define mx(p) t[p].mx
#define cnt(p) t[p].cnt
#define ls l(p),l,mid
#define rs r(p),mid+1,r
using namespace std;
char buf[1<<15],*fs,*ft;
inline char getc()
{
    return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;
}
inline int read()
{
    RE int x=0,f=1;RE char ch=getc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getc();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getc();}
    return x*f;
}
const int MAXN=200010;
int n,rt,id,flag,cnt,ww,cc,vv;
int b[10];
map<int,int>H;
set<int>s;
set<int>::iterator it;
struct wy
{
    int l,r;
    int mx,cnt;
}t[MAXN*30];
inline void insert(int &p,int l,int r,int x,int w)
{
    if(!p)p=++id;
    if(l==r)return mx(p)=w,void();
    int mid=(l+r)>>1;
    if(x<=mid)insert(ls,x,w);
    else insert(rs,x,w);
    mx(p)=min(mx(l(p)),mx(r(p)));
}
inline void modify(int p,int l,int r,int x)
{
    if(l==r)return cnt(p)^=1,void();
    int mid=(l+r)>>1;
    if(x<=mid)modify(ls,x);
    else modify(rs,x);
    cnt(p)=cnt(l(p))|cnt(r(p));
}
inline void insert(int x)
{
    ++H[x];++cc;ww=H[x];
    if(ww==1)
    {
        s.insert(x);
        it=s.find(x);
        ++it;
        if(it!=s.end())insert(rt,1,maxx,x,*it-x);
        else insert(rt,1,maxx,x,maxx+1);
        --it;
        if(it!=s.begin())
        {
            --it;
            insert(rt,1,maxx,*it,x-*it);
        }
    }
    if(ww==2)modify(rt,1,maxx,x);
}
inline void erase(int x)
{
    --H[x];--cc;ww=H[x];
    if(ww==0)
    {
        it=s.find(x);
        ++it;
        int qq=-1,hj=-1;
        if(it!=s.end())qq=*it;
        --it;
        if(it!=s.begin())--it,hj=*it;
        insert(rt,1,maxx,x,maxx+1);
        if(hj!=-1)insert(rt,1,maxx,hj,qq!=-1?qq-hj:maxx+1);
        s.erase(x);
    }
    if(ww==1)modify(rt,1,maxx,x);
}
inline void ask(int p,int l,int r,int L,int R)
{
    if(flag)return;
    if(L<=l&&R>=r)return flag=cnt(p)||(mx(p)<L),void();
    int mid=(l+r)>>1;
    if(L<=mid)ask(ls,L,R);
    if(R>mid)ask(rs,L,R);
}
inline int ask(int x)
{
    if(cc<=1)return 0;
    ww=H[x];flag=cnt=0;
    if(ww>=2)return 1;
    if(ww==1)b[cnt=1]=x;
    s.insert(x);it=s.find(x);
    if(it!=s.begin())
    {
        --it;
        ww=H[*it];
        rep(1,min(ww,2),i)b[++cnt]=*it;
        if(ww==1)
        {
            if(it!=s.begin())
            {
                --it;
                b[++cnt]=*it;
                ++it;
            }
        }
        ++it;
    }
    ++it;
    if(it!=s.end())b[++cnt]=*it;
    s.erase(x);
    rep(1,cnt,i)rep(i+1,cnt,j)
    {
        int cw=max(b[i],max(b[j],x));
        if(b[i]+b[j]>cw&&b[i]+x>cw&&b[j]+x>cw)return 1;
    }
    ask(rt,1,maxx,x,maxx);
    return flag;
}
int main()
{
    //freopen("1.in","r",stdin);
    get(n);mx(0)=maxx+1;
    rep(1,n,i)
    {
        int get(op),get(x);
        if(op==1)insert(x);
        if(op==2)erase(x);
        if(op==3)if(ask(x))puts("Yes");
        else puts("No");
    }
    return 0;
}
原文地址:https://www.cnblogs.com/chdy/p/13308782.html