10933 | 2024-04-20 15:56:34 | k_balint | Gyors utak | cpp17 | Elfogadva 100/100 | 207ms | 41476 KiB |
#include <bits/stdc++.h>
using namespace std;
const int c=2e5+5;
int tree[4*c][2], lazy[4*c];
int n,q,tt;
vector<int> adj[c];
int tin[c],tout[c],arr[c], mp[c];
void dfs(int v, int p){
arr[v]^=arr[p];
tin[v]=tout[v]=++tt;
mp[tin[v]]=v;
for(int x:adj[v]){
if(x != p){
dfs(x,v);
tout[v]=tout[x];
}
}
}
void build(int v, int l, int r){
if(l==r){
int x=mp[l];
tree[v][arr[x]]=1;
return;
}
int mid=l+r>>1;
build(2*v,l,mid);
build(2*v+1,mid+1,r);
tree[v][0]=tree[2*v][0]+tree[2*v+1][0];
tree[v][1]=tree[2*v][1]+tree[2*v+1][1];
}
void push(int v, int l, int r){
if(lazy[v]==0) return;
swap(tree[v][0],tree[v][1]);
if(l!=r){
lazy[2*v]^=1;
lazy[2*v+1]^=1;
}
lazy[v]=0;
}
void update(int v, int l, int r, int x, int y){
push(v,l,r);
if(l>r || l>y || x>r || x>y) return;
if(x<=l && r<=y){
swap(tree[v][0], tree[v][1]);
if(l!=r){
lazy[2*v]^=1;
lazy[2*v+1]^=1;
}
return;
}
int mid=l+r>>1;
update(2*v,l,mid,x,y);
update(2*v+1,mid+1,r,x,y);
tree[v][0]=tree[2*v][0]+tree[2*v+1][0];
tree[v][1]=tree[2*v][1]+tree[2*v+1][1];
}
int main(){
ios_base::sync_with_stdio(false);
cin.tie(0);
tt=0;
cin>>n>>q;
for(int i=2;i<=n;i++){
int a; cin>>a;
adj[a].push_back(i);
adj[i].push_back(a);
}
for(int i=2;i<=n;i++) cin>>arr[i];
dfs(1,0);
build(1,1,n);
cout << 1ll*tree[1][0]*(tree[1][0]-1)/2 + 1ll*tree[1][1]*(tree[1][1]-1)/2 << ' ';
while(q--){
int x; cin>>x;
++x;
update(1,1,n,tin[x],tout[x]);
cout << 1ll*tree[1][0]*(tree[1][0]-1)/2 + 1ll*tree[1][1]*(tree[1][1]-1)/2 << ' ';
}
}
Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
---|---|---|---|---|---|---|---|
subtask1 | 0/0 | ||||||
1 | Elfogadva | 6ms | 11344 KiB | ||||
2 | Elfogadva | 8ms | 12460 KiB | ||||
subtask2 | 10/10 | ||||||
3 | Elfogadva | 6ms | 12148 KiB | ||||
4 | Elfogadva | 6ms | 11896 KiB | ||||
5 | Elfogadva | 6ms | 12108 KiB | ||||
6 | Elfogadva | 6ms | 12116 KiB | ||||
7 | Elfogadva | 6ms | 12300 KiB | ||||
8 | Elfogadva | 6ms | 12256 KiB | ||||
subtask3 | 11/11 | ||||||
9 | Elfogadva | 6ms | 12148 KiB | ||||
10 | Elfogadva | 6ms | 11896 KiB | ||||
11 | Elfogadva | 6ms | 12108 KiB | ||||
12 | Elfogadva | 6ms | 12116 KiB | ||||
13 | Elfogadva | 6ms | 12300 KiB | ||||
14 | Elfogadva | 6ms | 12256 KiB | ||||
15 | Elfogadva | 8ms | 12992 KiB | ||||
16 | Elfogadva | 8ms | 12992 KiB | ||||
17 | Elfogadva | 8ms | 13048 KiB | ||||
18 | Elfogadva | 8ms | 13056 KiB | ||||
19 | Elfogadva | 8ms | 12992 KiB | ||||
20 | Elfogadva | 8ms | 13256 KiB | ||||
21 | Elfogadva | 9ms | 13236 KiB | ||||
22 | Elfogadva | 9ms | 13464 KiB | ||||
23 | Elfogadva | 9ms | 13544 KiB | ||||
24 | Elfogadva | 9ms | 13464 KiB | ||||
subtask4 | 21/21 | ||||||
25 | Elfogadva | 187ms | 41060 KiB | ||||
26 | Elfogadva | 187ms | 40940 KiB | ||||
27 | Elfogadva | 189ms | 41028 KiB | ||||
28 | Elfogadva | 193ms | 41228 KiB | ||||
29 | Elfogadva | 187ms | 41336 KiB | ||||
30 | Elfogadva | 187ms | 41476 KiB | ||||
subtask5 | 24/24 | ||||||
31 | Elfogadva | 7ms | 13388 KiB | ||||
32 | Elfogadva | 14ms | 15352 KiB | ||||
33 | Elfogadva | 108ms | 29108 KiB | ||||
34 | Elfogadva | 180ms | 29200 KiB | ||||
subtask6 | 34/34 | ||||||
35 | Elfogadva | 6ms | 12148 KiB | ||||
36 | Elfogadva | 6ms | 11896 KiB | ||||
37 | Elfogadva | 6ms | 12108 KiB | ||||
38 | Elfogadva | 6ms | 12116 KiB | ||||
39 | Elfogadva | 6ms | 12300 KiB | ||||
40 | Elfogadva | 6ms | 12256 KiB | ||||
41 | Elfogadva | 8ms | 12992 KiB | ||||
42 | Elfogadva | 8ms | 12992 KiB | ||||
43 | Elfogadva | 8ms | 13048 KiB | ||||
44 | Elfogadva | 8ms | 13056 KiB | ||||
45 | Elfogadva | 8ms | 12992 KiB | ||||
46 | Elfogadva | 8ms | 13256 KiB | ||||
47 | Elfogadva | 9ms | 13236 KiB | ||||
48 | Elfogadva | 9ms | 13464 KiB | ||||
49 | Elfogadva | 9ms | 13544 KiB | ||||
50 | Elfogadva | 9ms | 13464 KiB | ||||
51 | Elfogadva | 187ms | 41060 KiB | ||||
52 | Elfogadva | 187ms | 40940 KiB | ||||
53 | Elfogadva | 189ms | 41028 KiB | ||||
54 | Elfogadva | 193ms | 41228 KiB | ||||
55 | Elfogadva | 187ms | 41336 KiB | ||||
56 | Elfogadva | 187ms | 41476 KiB | ||||
57 | Elfogadva | 7ms | 13388 KiB | ||||
58 | Elfogadva | 14ms | 15352 KiB | ||||
59 | Elfogadva | 108ms | 29108 KiB | ||||
60 | Elfogadva | 180ms | 29200 KiB | ||||
61 | Elfogadva | 184ms | 29432 KiB | ||||
62 | Elfogadva | 180ms | 30244 KiB | ||||
63 | Elfogadva | 186ms | 30032 KiB | ||||
64 | Elfogadva | 181ms | 30084 KiB | ||||
65 | Elfogadva | 192ms | 29496 KiB | ||||
66 | Elfogadva | 188ms | 29512 KiB | ||||
67 | Elfogadva | 194ms | 29540 KiB | ||||
68 | Elfogadva | 202ms | 29564 KiB | ||||
69 | Elfogadva | 165ms | 29272 KiB | ||||
70 | Elfogadva | 197ms | 29640 KiB | ||||
71 | Elfogadva | 157ms | 29064 KiB | ||||
72 | Elfogadva | 200ms | 30108 KiB | ||||
73 | Elfogadva | 207ms | 30624 KiB | ||||
74 | Elfogadva | 127ms | 30628 KiB | ||||
75 | Elfogadva | 192ms | 29484 KiB | ||||
76 | Elfogadva | 193ms | 29492 KiB | ||||
77 | Elfogadva | 178ms | 30516 KiB |