比赛 CSP2023-S模拟赛 评测结果 WWWWWWWWWWWWWWWWWWWW
题目名称 删除题目 最终得分 0
用户昵称 嗨嗨嗨 运行时间 0.419 s
代码语言 C++ 内存使用 13.94 MiB
提交时间 2023-10-17 20:49:06
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
int n,x,y,st[20][100010];
unsigned long long ans;
struct node
{
    int fa,w,depth,up;
    unsigned long long num,vis;
    vector<int>c;
}t[100010];
inline int read()
{
	int k=0,f=1;
    char c=getchar();
	for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
	for(;isdigit(c);c=getchar()) k=(k<<1)+(k<<3)+c-'0';
	return k*f;
}
inline void write(int x)
{
    if(x<0) putchar('-'),x=-x;
    if(x>9) write(x/10);
    putchar(x%10+'0');
}
inline void dfs1(int p,int f)
{
    t[p].num++;
    t[p].depth=t[f].depth+1;
    st[0][p]=f;
    for(int i=1;i<=19;i++) st[i][p]=st[i-1][st[i-1][p]];
    for(int i=0;i<t[p].c.size();i++)
    {
        int j=t[p].c[i];
        dfs1(j,p);
        t[p].num+=t[j].num;
    }
    t[p].vis=t[p].num;
}
inline int LCA(int x,int y)
{
    if(t[x].depth<t[y].depth) swap(x,y);
    int h=t[x].depth-t[y].depth;
    for(int i=19;i>=0;i--) if(h>=(1<<i)) h-=1<<i,x=st[i][x];
    if(x==y) return x;
    for(int i=19;i>=0;i--) if(st[i][x]!=st[i][y]) x=st[i][x],y=st[i][y];
    return st[0][x];
}
inline void dfs2(int p,int las)
{
    if(p==1) t[p].up=1;
    else t[p].up=t[t[p].fa].up;
    int lca=LCA(p,las),flag=las;
    unsigned long long tmp;
    if(p!=1)
    {
        tmp=t[p].num*(t[t[lca].up].vis-t[p].num);
        if(tmp>t[p].w)
        {
            las=p;
            ans+=tmp-t[p].w;
            flag=p;
            t[p].up=p;
            t[t[p].fa].vis-=t[p].num;
            cout<<p<<" ";
        }
    }
    for(int i=0;i<t[p].c.size();i++)
    {
        int j=t[p].c[i];
        dfs2(j,flag);
    }
}
int main()
{
    freopen("delete.in","r",stdin);
    freopen("delete.out","w",stdout);
    n=read();
    for(int i=2;i<=n;i++)
    {
        t[i].fa=read(),t[i].w=read();
        t[t[i].fa].c.push_back(i);
    }
    dfs1(1,0);
    dfs2(1,0);
    write(ans);
    return 0;
}