LOJ.2865.[IOI2018]狼人(Kruskal重构树 主席树)

LOJ
洛谷


这题不就是Peaks(加强版)或者归程么。。这算是\(IOI2018\)撞上\(NOI2018\)的题了?
\(Kruskal\)重构树(具体是所有点按从小到大/从大到小的顺序,依次加入这些点的边),我们可以得到两棵树(和那两题不一样的是这题的权值在点上,不需要新建节点)。
对于询问\((S,T,L,R)\),可以倍增找出\(S,T\)可以在哪棵子树中随便走。
那么只需要判断两棵子树是否有交就可以惹。
注意到子树的DFS序是连续的,我们可以在第一个子树的某个数据结构上,查第二个子树的DFS序区间。用主席树就可以了,主席树的下标是每个点在第二棵树上的DFS序。判断第一个子树在第二个子树的DFS序区间中是否有值即可。

写一写权当复习惹。
另外,这个题的交互有点逗逼啊...为什么是所有询问一起输出啊?


LOJ上15分,T掉了,不知道为什么(应该是交互的问题)。不管惹。

//656ms 84.03MB
#include <cstdio>
#include <cctype>
#include <algorithm>
#define BIT 17
#define gc() getchar()
#define MAXIN 300000
//#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=2e5+5,M=8e5+5;

int n,root[N];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Graph
{
    int Enum,H[N],nxt[M],to[M];
    inline void AE(int u,int v)
    {
        to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
        to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum;
    }
}G;
struct Tree
{
    int fg,fa[N][BIT+1],F[N],H[N],nxt[N],L[N],R[N],Index;
    int Find(int x)
    {
        return x==F[x]?x:F[x]=Find(F[x]);
    }
    inline void AE(int u,int v)
    {
        nxt[v]=H[u], H[u]=v;
    }
    inline int Jump(int x,int k)
    {
        if(fg)
        {
            for(int i=BIT; ~i; --i)
                if(fa[x][i]&&fa[x][i]<=k) x=fa[x][i];
        }
        else
        {
            for(int i=BIT; ~i; --i)
                if(fa[x][i]>=k) x=fa[x][i];
        }
        return x;
    }
    void DFS(int x,int dep)
    {
        L[x]=++Index;
        for(int i=1; 1<<i<=dep; ++i) fa[x][i]=fa[fa[x][i-1]][i-1];
        for(int v=H[x]; v; v=nxt[v]) fa[v][0]=x, DFS(v,dep+1);
        R[x]=Index;
    }
    void Build(const int n)
    {
        for(int i=1; i<=n; ++i) F[i]=i;
        if(fg)
        {
            for(int x=1,k=1; x<=n; ++x)
                for(int i=G.H[x],v; i; i=G.nxt[i])
                    if((v=G.to[i])<x)
                    {
                        int r1=Find(x),r2=Find(v);//r1==x
                        if(r1!=r2)
                        {
                            F[r2]=r1, AE(r1,r2);
                            if(++k==n) break;
                        }
                    }
        }
        else
        {
            for(int x=n,k=1; x; --x)
                for(int i=G.H[x],v; i; i=G.nxt[i])
                    if((v=G.to[i])>x)
                    {
                        int r1=Find(x),r2=Find(v);
                        if(r1!=r2)
                        {
                            F[r2]=r1, AE(r1,r2);
                            if(++k==n) break;
                        }
                    }
        }
        DFS(fg?n:1,0);
    }
}T1,T2;
struct Segment_Tree
{
    #define ls son[x][0]
    #define rs son[x][1]
    #define lson ls,son[y][0],l,m
    #define rson rs,son[y][1],m+1,r
    #define S N*19
    int tot,sz[S],son[S][2];
    #undef S
    void Modify(int &x,int y,int l,int r,int p)
    {
        sz[x=++tot]=sz[y]+1;
        if(l==r) return;
        int m=l+r>>1;
        p<=m?(rs=son[y][1],Modify(lson,p)):(ls=son[y][0],Modify(rson,p));
    }
    int Query(int x,int y,int l,int r,int L,int R)//y-x
    {
        if(sz[y]-sz[x]<=0) return 0;
        if(L<=l && r<=R) return 1;
        int m=l+r>>1;
        if(L<=m)
            if(m<R) return Query(lson,L,R)||Query(rson,L,R);
            else return Query(lson,L,R);
        return Query(rson,L,R);
    }
}T;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-48,c=gc());
    return now;
}
void DFS(int x)
{
    static int Index=0;
    ++Index, T.Modify(root[Index],root[Index-1],1,n,T2.L[x]);
    for(int v=T1.H[x]; v; v=T1.nxt[v]) DFS(v);
}

int main()
{
    int n=read(),m=read(),Q=read(); ::n=n;
    for(int i=1; i<=m; ++i) G.AE(read()+1,read()+1);
    T1.Build(n), T2.fg=1, T2.Build(n), DFS(1);
    for(int s,t; Q--; )
        s=read()+1,t=read()+1,s=T1.Jump(s,read()+1),t=T2.Jump(t,read()+1),
        putchar(T.Query(root[T1.L[s]-1],root[T1.R[s]],1,n,T2.L[t],T2.R[t])?'1':'0'), putchar('\n');

    return 0;
}

转载于:https://www.cnblogs.com/SovietPower/p/10685264.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值