【HNOI2010】弹飞绵羊(LCT)

这道题不用换根,因为换根存在的意义就是方便查询路径(u,v)上的信息,而这道题只需要一个点的size

#include<bits/stdc++.h>
const int N=200005;
using namespace std;
template<class T>
inline void read(T &x)
{
    x=0; int f=1;
    static char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
    x*=f;
}
int n,d[N],m;
struct Link_Cut_Tree
{
    int lc,rc,size,fa;
}t[N];
inline void pushup(int u){t[u].size=t[t[u].lc].size+t[t[u].rc].size+1;}
inline int which(int u){return t[t[u].fa].rc==u;}
inline bool isroot(int u){return t[t[u].fa].lc!=u&&t[t[u].fa].rc!=u;}
inline void Rotate(int u)
{
    int v=t[u].fa,w=t[v].fa;
    int b=(t[v].rc==u)?t[u].lc:t[u].rc;
    if(!isroot(v)) (t[w].lc==v?t[w].lc:t[w].rc)=u;
    t[u].fa=w; t[v].fa=u;
    if(b) t[b].fa=v;
    if(t[v].rc==u)	t[v].rc=b,t[u].lc=v;
    else t[v].lc=b,t[u].rc=v;
    pushup(v); pushup(u);
}
inline void Splay(int u)
{
    while(!isroot(u))
    {
        if(!isroot(t[u].fa))
        {
            if(which(u)==which(t[u].fa))	Rotate(t[u].fa);
            else Rotate(u);
        }
        Rotate(u);
    }
}
inline void access(int u)
{
    for(int v=0;u;u=t[v=u].fa)
        Splay(u),t[u].rc=v,pushup(u);
}
inline void Link(int u,int v){ Splay(u); t[u].fa=v;} //刚刚access过了 
inline void Cut(int u)
{
    access(u); Splay(u);
    t[t[u].lc].fa=0; t[u].lc=0; pushup(u);
}
inline void write(int x)
{
	if(x<0){x=-x; write(x);}
	if(x>9) write(x/10);
	putchar(x%10+'0');
}
int main()
{
    read(n);
    for(int i=1;i<=n;i++)
    {
        read(d[i]); t[i].size=1;
        if(i+d[i]<=n) Link(i,i+d[i]);
    }
    read(m);
    for(int i=1,opt,x,y;i<=m;i++)
    {
        read(opt); read(x); x++;
        if(opt==1) 
        {
            access(x); Splay(x);
            write(t[x].size),putchar('\n');
        }
        else
        {
            read(y); Cut(x); d[x]=y;
            if(x+d[x]<=n) Link(x,x+d[x]);
        }
    }
    return 0;
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值