/ SeriousOJ /

Record Detail

Compile Error


  

Code

#include<bits/stdc++.h>
using namespace std;
const long long M=2e5+10,MOD=1000000007;
typedef long long ll;
vector<int>edge[M];
int in[M],out[M];
vector<int>index;
int bit[M];
int cnt=1;
int query(int id){
    int sum=0;
    while(id>0){
         sum+=bit[id];
         id-=id&(-id);
    }
    return sum;
}
void upade(int id,int val,int n){
           while(id<=n){
            bit[id]+=val;
            id+=id&(-id);
           }
}
void dfs(int x,int p){
    in[x]=cnt;
    index.push_back(x);
    for(int u:edge[x]){
        if(u!=p){
            dfs(u,x);
        }
    }
    cnt++;
    out[x]=cnt;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t=1;
    cin>>t;
    int cas=1;
    while(t--){
     
      int n,q;
      cin>>n>>q;
      vector<int>a(n+1);
      for(int i=1;i<=n;i++)cin>>a[i];
      for(int i=1;i<n;i++){
        int x,y;
        cin>>x>>y;
        edge[x].push_back(y);
        edge[y].push_back(x);
      }
      cnt=1;
      dfs(1,-1);
      while(q--){
        int node;
        cin>>node;
        upade(in[node],1,n);
      }
      for(int i=0;i<n;i++){
        int x=index[i];
        int cnt=query(in[x])-query(in[x]-1);
        if(cnt&1){
            if(a[x]==0)a[x]=1;
            else a[x]=0;
        }
      }
      index.clear();
    //  cas++;
      cout<<"Case "<<cas<<": ";
      for(int i=1;i<=n;i++)cout<<a[i]<<" ";
        cout<<"\n";
     for(int i=1;i<=n;i++){
        edge[i].clear();
        in[i]=0;
        out[i]=0;
        bit[i]=0;
     }
     cas++;

   }
       

    
    return 0;
 
}

Information

Submit By
Type
Pretest
Problem
P1003 Tahsin and Tree
Language
C++17 (G++ 13.2.0)
Submit At
2023-11-29 17:38:48
Judged At
2023-11-29 20:47:07
Judged By
Score
0
Total Time
0ms
Peak Memory
0 Bytes