接水果【题解】

1 篇文章 0 订阅
1 篇文章 0 订阅

前言

虽然是NOI-/NOI/CTSC,但感觉还没有平时考试难,可能是这种题太套路了.

题面

给了你一棵树,给了你一些路径,每次询问,u,v两点间的第k小的子路径(必须是给出的那些路径中的)

sol

怎么判某条路径是某条路径的子路径?
一个套路就是用dfs序的区间包含关系.这个套路大概自己推理一下就出来了.
然后发现限制条件是一个二维的,所以这个区间不是一个线段,而是一个矩形,所以用一下扫描线.当然可以用树套树,二维树状数组,但是他们大材小用。注意到不需要修改,而这些数据结构支持修改,所以不需要用。
扫描线就是把一个矩形拆成两条线段,分别是平行于y轴的两条.每个线段记录一下横坐标,记录一下两个纵坐标,然后在记录一下是左边的还是右边的。然后按x排序然后顺次加入,计入左边的就给y1-y2加一,计入右边的就减一.

code

// luogu-judger-enable-o2
#include<bits/stdc++.h>
using namespace std;
template <class T>
inline void read(T&data){
    register char ch=0;
    data=0;
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch<='9'&&ch>='0'){
        data=(data<<3)+(data<<1)+(ch^48);
        ch=getchar();
    }
    return;
}
#define rg register
const int _ = 100011;
struct edge {
    int nt,to;
}e[_<<1];
int cnt=0,n,head[_],m,q,son[_],size[_],top[_],fa[_],deep[_],Index,dfn[_],low[_];
int Map[_],ans[_];
struct Line{
    int ki,x,y1,y2,d,val;
}Q[_<<2],Q1[_<<2],Q2[_<<2];
int tree[_<<2];
inline int lowbit(register int k){return k&(-k);}
inline void modify(register int loc,register int zh){for(register int i=loc;i<=Index+1;i+=lowbit(i))tree[i]+=zh;}
inline int query(register int loc){register int ret=0;for(register int i=loc;i;i-=lowbit(i))ret+=tree[i];return ret;}
void add(register int a,register int b){e[++cnt].to=a,e[cnt].nt=head[b],head[b]=cnt;}
bool cmp(Line a,Line b){return a.val<b.val;}
bool cmp2(Line a,Line b){if(a.x!=b.x)return a.x<b.x;return a.ki<b.ki;}
void dfsI(register int now,register int ff){
    deep[now]=deep[ff]+1,fa[now]=ff,size[now]=1;
    for(register int i=head[now];i;i=e[i].nt)if(e[i].to!=ff){
            dfsI(e[i].to,now);size[now]+=size[e[i].to];
            if(size[e[i].to]>size[son[now]])son[now]=e[i].to;
        }
}
void dfsII(register int now,register int topf){
    dfn[now]=++Index,top[now]=topf;
    if(son[now])dfsII(son[now],topf);
    for(register int i=head[now];i;i=e[i].nt)if(dfn[e[i].to]==0)dfsII(e[i].to,e[i].to);
    low[now]=++Index;
}
inline int getlca(register int a,register int b,register int &yl){
    while(top[a]^top[b]){
        if(deep[top[a]]<deep[top[b]]) swap(a,b);
        yl=top[a],a=fa[top[a]];
    }
    if(deep[a]<deep[b])swap(a,b);
    if(a!=b)
    yl=son[b];
    return b;
}
inline void Insert(register int a,register int b,register int c,register int d,register int e,register int f){
        Q[++cnt].ki=a,Q[cnt].x=b,Q[cnt].y1=c,Q[cnt].y2=e,Q[cnt].val=f,Q[cnt].d=1;
        Q[++cnt].ki=a,Q[cnt].x=d+1,Q[cnt].y1=c,Q[cnt].y2=e,Q[cnt].val=f,Q[cnt].d=-1;
}
void divide(register int l,register int r,register int L,register int R){
    //cout<<l<<' '<<r<<endl;
    if(l==r){
        for(register int i=L;i<=R;++i)
            if(Q[i].ki==2)
                ans[Q[i].y2]=l;
        return;
    }
    register int mid = (l+r)>>1;
    register int t1=0,t2=0;
    for(register int i=L;i<=R;++i){
        if(Q[i].ki==1){
            if(Q[i].val<=mid)   Q1[++t1]=Q[i],modify(Q[i].y1,Q[i].d),modify(Q[i].y2+1,-Q[i].d);
            else Q2[++t2]=Q[i];
        }
        else {
            register int su=query(Q[i].y1);
            if(Q[i].d<=su){Q1[++t1]=Q[i];/*cout<<Q[i].y2<<' '<<mid<<'l'<<endl;*/}
            else Q[i].d-=su,Q2[++t2]=Q[i]/*,cout<<Q[i].y2<<' '<<mid<<'r'<<endl*/;
        }
    }
    for(register int i=1;i<=t1;++i)Q[i+L-1]=Q1[i];
    for(register int i=1;i<=t2;++i)Q[t1+L-1+i]=Q2[i];
    divide(l,mid,L,L+t1-1),divide(mid+1,r,L+t1,R);return;
}
int main(){
    //freopen("data.in","r",stdin);
    //freopen("1.out","w",stdout);
    register int a,b,c;
    read(n),read(m),read(q);
    for(rg int i=1;i<n;++i){
        read(a),read(b);
        add(a,b),add(b,a);
    }
    dfsI(1,0);dfsII(1,1);cnt=0;
    for(register int i=1;i<=m;++i){
        read(a),read(b),read(c);
        if(dfn[a]>dfn[b])swap(a,b);
        register int yl=0,lca=getlca(a,b,yl);//cout<<yl<<endl;
        if(lca==a){
            if(dfn[yl]>=1)Insert(1,dfn[b],1,low[b],dfn[yl]-1,c);
            //if(dfn[yl]<Index)Insert(1,dfn[b],low[yl],low[b],Index,c);
            if(yl)Insert(1,low[yl],dfn[b],Index,low[b],c);
        }
        else Insert(1,dfn[b],dfn[a],low[b],low[a],c);
    }
    //exit(0);
    /*for(register int i=1;i<=cnt;++i){
        cout<<Q[i].x<<' '<<Q[i].y1<<' '<<Q[i].y2<<endl;
        }*/
    a=-1,b=0;
    sort(Q+1,Q+cnt+1,cmp);
    for(register int i=1;i<=cnt;++i){
        if(Q[i].val==a) Q[i].val=b;
        else Map[++b]=Q[i].val,a=Q[i].val,Q[i].val=b;
    }
    //cout<<endl;
    register int yl=b;
    for(register int i=1;i<=q;++i){
        read(a),read(b),read(c);
        if(dfn[a]>dfn[b])swap(a,b);
        Q[++cnt].ki=2,Q[cnt].x=dfn[b],Q[cnt].y1=dfn[a],Q[cnt].y2=i,Q[cnt].d=c;
    }
    sort(Q+1,Q+cnt+1,cmp2);b=yl;
    divide(1,b,1,cnt);
    for(register int i=1;i<=q;++i){
        cout<<Map[ans[i]]<<endl;
    }
}
/*
5 4 4
1 2
2 3
3 4
4 5
1 2 10
2 4 11
3 4 12
3 5 13
1 3 2
1 3 1
1 4 4
1 5 3
 */
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值