记录编号 197036 评测结果 AAAAAAAAAA
题目名称 [河南省队2012] 座位问题 最终得分 100
用户昵称 Gravatardevil 是否通过 通过
代码语言 C++ 运行时间 0.282 s
提交时间 2015-10-23 10:57:42 内存使用 9.47 MiB
显示代码纯文本
#include <cstdlib>
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <map>
#include <stack>
#include <vector>
#include <map>
#include <queue>
#include <ctime>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef unsigned int uint;
const int inf=1061109567;
const int maxn=200010;
const int maxm=10000010;
const int mod=1000000000;

inline int Read()
{
    int x=0;char y;
    do{y=getchar();} while(y<'0'||y>'9');
    do{x=x*10+y-'0';y=getchar();} while(y>='0'&&y<='9');
    return x;
}

struct node
{
    int val;
    int add;
    node() {val=add=0;}
} tree[maxn*3];

int first[maxn],next[maxn];
//int son[maxn],pos[maxn];
int u[maxn],v[maxn];
//int s[maxn],t[maxn];
int vis[maxn],f[maxn],ans;

void add_G(int u,int v,int c) {next[c]=first[u];first[u]=c;}

/*int build_tree(int root)
{
    int c=1;
    vis[root]=1;
    for(int i=first[root];i!=-1;i=next[i])
    {
        if(!vis[v[i]])
        {
            c+=build_tree(v[i]);
            son[root]=pos[v[i]];
        }
    }
    pos[root]=cnt++;
    return s[root]=c;
}

void Add(int al,int ar,int l,int r,int dir)
{
    if(al>r||ar<l) return;
    if(al<=l&&ar>=r) {tree[dir].add++;return;}
    int m=(l+r)>>1;
    Add(al,ar,l,m,dir*2);
    Add(al,ar,m+1,r,dir*2+1);
}

int query(int pos,int l,int r,int dir,int sum)
{
    if(l==r) return sum+tree[dir].add;
    int m=(l+r)>>1;
    if(pos<=m) return query(pos,l,m,dir*2,sum+tree[dir].add);
    else return query(pos,m+1,r,dir*2+1,sum+tree[dir].add);
}*/

void build_tree(int root)
{
    vis[root]=1;
    for(int i=first[root];i!=-1;i=next[i])
    {
        if(!vis[v[i]])
        {
            build_tree(v[i]);
            f[v[i]]=root;
        }
    }
}

void query(int pos)
{
    if(pos==0) return;
    if(vis[pos]) ans++;
    query(f[pos]);
}

int main()
{
    freopen("seat.in","r",stdin);
    freopen("seat.out","w",stdout);
    //clock_t st=clock();
    memset(first,-1,sizeof(first));
    int n;n=Read();
    int i,x;
    for(i=1;i<n;i++)
    {
        u[i]=Read();v[i]=Read();
        add_G(u[i],v[i],i);
        u[i+n-1]=v[i];v[i+n-1]=u[i];
        add_G(u[i+n-1],v[i+n-1],i+n-1);
    }
    build_tree(1);
    //for(int i=1;i<=n;i++) printf("%d %d\n",i,s[i]);
    memset(vis,0,sizeof(vis));
    for(i=1;i<=n;i++)
    {
        x=Read();
        /*printf("%d\n",query(pos[x],1,n,1,0));
        if(son[x]!=0) Add(pos[x]-s[x]+1,pos[x],1,n,1);*/
        ans=0;query(x);vis[x]=1;
        printf("%d\n",ans);
    }
    //clock_t ed=clock();
    //printf("\nTime used : %.5lf Ms\n",double(ed-st)/CLOCKS_PER_SEC);
    return 0;
}