记录编号 600505 评测结果 AAAAAAAAAA
题目名称 终末鸟 最终得分 100
用户昵称 Gravatardjyqjy 是否通过 通过
代码语言 C++ 运行时间 17.426 s
提交时间 2025-05-05 15:53:22 内存使用 78.53 MiB
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
inline int re()
{
    int f=1,num=0;
    char c=getchar();
    while(c<'0'||c>'9'){if(c=='-') f=-1;c=getchar();}
    while(c>='0'&&c<='9') num=num*10+c-'0',c=getchar();
    return num*f;
}
const int N=10000010;
int n,q;
int ver[2*N],nxt[2*N],hd[N],f[N],d[N];
int jsq=1;
void add_edge(int x,int y)
{
    ver[++jsq]=y;
    nxt[jsq]=hd[x];
    hd[x]=jsq;
    ver[++jsq]=x;
    nxt[jsq]=hd[y];
    hd[y]=jsq;
    return;
}
bool mark[N];
void dfs(int x,int fa)
{
    f[x]=fa;
    for(int i=hd[x];i;i=nxt[i])
    {
        int y=ver[i];
        if(y==fa) continue;
        dfs(y,x);
    }
    return;
}
int ans;
void insert(int x)
{
    if(mark[x]==0)
    {
        ans-=d[x]+mark[f[x]]-1;
        d[f[x]]++;
    }
    else
    {
        ans+=d[x]+mark[f[x]]-1;
        d[f[x]]--;
    }
    mark[x]^=1;
    return;
}
int main()
{
    freopen("birds.in","r",stdin);
    freopen("birds.out","w",stdout);
    n=re();
    for(int i=1,x,y;i<n;i++) x=re(),y=re(),add_edge(x,y);
    dfs(1,0);
    for(int i=1;i<=n;i++) if(re()) insert(i);
    printf("%d\n",ans);
    q=re();
    for(int i=1;i<=q;i++) insert(re()),printf("%d\n",ans);
    return 0;
}