比赛 2026.4.11 评测结果 AAAAAAAAAA
题目名称 粒子对撞 最终得分 100
用户昵称 RpUtl 运行时间 7.119 s
代码语言 C++ 内存使用 32.08 MiB
提交时间 2026-04-11 12:38:33
显示代码纯文本
#include <bits/stdc++.h>
#include "nuclear.h" 
using namespace std;
const int N=2e5+10;
vector<int>G[N];
int siz[N],Fa[N],de[N],son[N],n;
int top[N],dfn[N],cnt,tmp,ans;
int mk[N],val[N],del[N];
void add(int x,int y){
    G[x].push_back(y);
}
struct node{
    int d,x;
    bool operator < (const node &u)const{
        return d<u.d;
    }
};
set<node>st[N];
void dfs1(int x,int fa){
    siz[x]=1,Fa[x]=fa;
    de[x]=de[fa]+1;
    for(auto y:G[x]){
        if(y==fa)continue;
        dfs1(y,x);
        siz[x]+=siz[y];
        if(siz[y]>siz[son[x]]){
            son[x]=y;
        }
    }
    return;
}
void dfs2(int x,int tp){
    top[x]=tp,dfn[x]=++cnt;
    if(!son[x])return;
    dfs2(son[x],tp);
    for(auto y:G[x]){
        if(y==Fa[x]||y==son[x])continue;
        dfs2(y,y);
    }
    return;
}
int found(int x){
    while(1){
        int tp=top[x];
        if(st[tp].size()){
            auto it=st[tp].upper_bound(node{de[x],0});
            if(it!=st[tp].begin()){
                --it;
                return (*it).x;
            }
        }
        x=Fa[tp];
    }
    return 1;
}
struct sgt{
    int tag[N<<2];
    #define ls (p<<1)
    #define rs (p<<1|1)
    void add(int p,int l,int r,int L,int R,int v){
        if(L<=l&&r<=R){
            tag[p]+=v;
        }else{
            int mid=(l+r)>>1;
            if(L<=mid)add(ls,l,mid,L,R,v);
            if(R>mid)add(rs,mid+1,r,L,R,v);
        }
    }
    int ask(int p,int l,int r,int x){
        if(l==r)return tag[p];int mid=(l+r)>>1;
        if(x<=mid)return ask(ls,l,mid,x)+tag[p];
        if(x>mid)return ask(rs,mid+1,r,x)+tag[p];
    }
}tr;
void upd(int a,int b,int c){
    while(top[a]!=top[b]){
        tr.add(1,1,n,dfn[top[a]],dfn[a],c);
        a=Fa[top[a]];
    }
    tr.add(1,1,n,dfn[b],dfn[a],c);
    return;
} 
int ask(int u){
    return tr.ask(1,1,n,dfn[u]);   
}
void initialize(int N, std::vector<int> A, std::vector<int> B){
    n=N;
    for(int i=0;i<n-1;i++){
        A[i]++,B[i]++;
        add(A[i],B[i]),add(B[i],A[i]);
    } 
    dfs1(1,0);
    dfs2(1,1);
    st[1].insert(node{1,1});
    mk[1]=1;
}
int generate(int u, bool result){
    u++;int x=result;
    if(x==1){
        int rot=found(u);
        tmp=ask(rot);
        ans-=tmp/2;
        upd(u,rot,1);
        tmp=ask(rot);
        ans+=tmp/2;
    }else{
        del[u]=1;
        int rot=found(u);
        int sz=ask(u);
        tmp=ask(rot);
        ans-=tmp/2;
        if(u!=rot){   
            upd(Fa[u],rot,-sz);
            tmp=ask(rot);
            ans+=tmp/2;
        }else{
            st[top[u]].erase(node{de[u],u});
        }
        for(auto v:G[u]){
            if(v==Fa[u])continue;
            if(del[v])continue;
            st[top[v]].insert(node{de[v],v});
            tmp=ask(v);
            ans+=tmp/2;
            mk[v]=1;
        }
    }
    return ans;
}