hdu 6039 Gear Up(dfs序+线段树)

题目链接：hdu 6039 Gear Up

题意：

给出一些齿轮，有些齿轮是边相连，也就是拥有相同的线速度，
有的齿轮是轴相连，也就是拥有相同的角速度，现在给某个齿轮一个速度，
求这些齿轮中的最大速度，同时还有修改操作，可以更改某个齿轮的半径大小

题解：

官方题解：

#include<bits/stdc++.h>
#define mst(a,b) memset(a,b,sizeof(a))
#define F(i,a,b) for(int i=(a);i<=(b);++i)
inline int RT(int l,int r){return l+r|l!=r;}
using namespace std;

const int N=1e5+7;
struct Node{
    int st,i,val;
    Node(int a,int b,int c):st(a),i(b),val(c){}
};
int g[N],v[N*2],nxt[N*2],ed,mx[N*4],lazy[N*4];
int n,m,q,r[N],f[N],is[N],Rt[N],sp[N*2];
int in[N],out[N],ll[N],rr[N],idx,a,b,c,cas;
vector<Node>G[N];

int find(int x){return f[x]!=x?f[x]=find(f[x]):x;}
void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}

void dfs(int x,int fa,int v,int rt)
{
    in[x]=++idx,Rt[x]=rt,sp[idx]=v;
    for(auto &it:G[x])
    {
        if(it.st!=fa)
        {
            ll[it.i]=min(ll[it.i],idx+1);
            dfs(it.st,x,v+it.val,rt);
            rr[it.i]=max(rr[it.i],idx);
        }else is[it.i]=1;//是否与父节点线连
    }
    out[x]=++idx,sp[idx]=v;
}

inline void PU(int l,int r)
{
    int mid=l+r>>1;
    mx[RT(l,r)]=max(mx[RT(l,mid)],mx[RT(mid+1,r)]);
}
inline void PD(int l,int r)
{
    int rt=RT(l,r),mid=l+r>>1;
    int ls=RT(l,mid),rs=RT(mid+1,r);
    mx[ls]+=lazy[rt],lazy[ls]+=lazy[rt];
    mx[rs]+=lazy[rt],lazy[rs]+=lazy[rt];
    lazy[rt]=0;
}

void build(int l=1,int r=idx)
{
    int rt=RT(l,r),mid=l+r>>1;
    mx[rt]=-INT_MAX,lazy[rt]=0;
    if(l==r){mx[rt]=sp[l];return;}
    build(l,mid),build(mid+1,r),PU(l,r);
}

void update(int L,int R,int v,int l=1,int r=idx)
{
    int rt=RT(l,r),mid=l+r>>1;
    if(L<=l&&r<=R){mx[rt]+=v,lazy[rt]+=v;return;}
    if(lazy[rt]!=0)PD(l,r);
    if(L<=mid)update(L,R,v,l,mid);
    if(R>mid)update(L,R,v,mid+1,r);
    PU(l,r);
}

int ask(int L,int R,int l=1,int r=idx)
{
    int rt=RT(l,r),mid=l+r>>1,ans=-INT_MAX;
    if(L<=l&&r<=R)return mx[rt];
    if(lazy[rt]!=0)PD(l,r);
    if(L<=mid)ans=max(ans,ask(L,R,l,mid));
    if(R>mid)ans=max(ans,ask(L,R,mid+1,r));
    return ans;
}

int main(){
    while(~scanf("%d%d%d",&n,&m,&q))
    {
        mst(g,0),ed=idx=0;
        F(i,1,n)
        {
            f[i]=i,ll[i]=INT_MAX;
            rr[i]=Rt[i]=is[i]=0;
            scanf("%d",r+i),G[i].clear();
            r[i]=__builtin_ctz(r[i]);
        }
        F(i,1,m)
        {
            scanf("%d%d%d",&a,&b,&c);
            if(a==1)adg(b,c),adg(c,b);
            else f[find(b)]=find(c);
        }
        
        F(i,1,n)for(int j=g[i];j;j=nxt[j])
        G[find(i)].push_back(Node(find(v[j]),i,r[i]-r[v[j]]));
        F(i,1,n)if(find(i)==i&&!Rt[i])dfs(i,0,0,i);
        build(),printf("Case #%d:
",++cas);
        while(q--)
        {
            scanf("%d%d%d",&a,&b,&c);
            c=__builtin_ctz(c);
            if(a==1)
            {
                int now=c-r[b];
                if(is[b])update(in[find(b)],out[find(b)],-now);
                if(ll[b]<=rr[b])update(ll[b],rr[b],now);
                r[b]=c;
            }else
            {
                b=find(b);
                int ans=c+ask(in[Rt[b]],out[Rt[b]])-ask(in[b],in[b]);
                printf("%.3f
",1.0*ans*log(2));
            }
        }
    }
    return 0;
}