10180 | 2024-03-29 10:39:00 | 111 | Különböző katicák | cpp17 | Wrong answer 0/100 | 64ms | 52456 KiB |
#include <bits/stdc++.h>
using namespace std;
#define int long long
signed main(){
ios::sync_with_stdio(false);
cin.tie(nullptr);
int N;
cin>>N;
vector<vector<int>>g(N+1);
vector<int>p(N+1);
for(int i=1;i<=N;i++){
cin>>p[i];
if(i==1){
continue;
}
g[p[i]].push_back(i);
g[i].push_back(p[i]);
}
vector<int>v(N+1);
for(int i=1;i<=N;i++){
cin>>v[i];
}
vector<pair<int,int>>w(N+1);
auto dfs=[&](auto self,int i)->pair<int,int>{
pair<int,int>a={v[i],v[i]};
for(int j:g[i]){
pair<int,int>b=j==p[i]?v[j]==-1?pair<int,int>{-1,-1}:pair<int,int>{abs(v[j]-1),v[j]+1}:self(self,j);
if(b.first==-1&&b.second==-1){
continue;
}
if(a.first==-1&&a.second==-1){
a=b;
continue;
}
if(a.first%2!=b.first%2||b.first>a.second||b.second<a.first){
cout<<"NEM"<<'\n';
exit(0);
}
a.first=max(a.first,b.first);
a.second=min(a.second,b.second);
}
w[i]=a;
if(a.first!=-1&&a.second!=-1){
a.first=a.first==0?a.first+1:a.first-1;
a.second++;
}
return a;
};
dfs(dfs,1);
auto dfs2=[&](auto self,int i)->void{
for(int j:g[i]){
if(j==p[i]){
continue;
}
if(v[j]==-1){
v[j]=v[i]+1;
}
else{
if(w[j].first<=v[i]+1&&v[i]+1<=w[j].second){
v[j]=v[i]+1;
}
else{
v[j]=v[i]-1;
}
}
self(self,j);
}
};
dfs2(dfs2,1);
for(int i=1;i<=N;i++){
for(int j:g[i])if(abs(v[i]-v[j])!=1){
cout<<i<<" "<<v[i]<<endl;
cout<<j<<" "<<v[j]<<endl;
exit(1);
}
}
cout<<"IGEN"<<'\n';
for(int i=1;i<=N;i++){
cout<<v[i]<<' ';
}
cout<<'\n';
return 0;
}
Subtask | Sum | Test | Verdict | Time | Memory | ||
---|---|---|---|---|---|---|---|
subtask1 | 0/0 | ||||||
1 | Accepted | 3ms | 1824 KiB | ||||
2 | Accepted | 3ms | 2060 KiB | ||||
3 | Wrong answer | 48ms | 19208 KiB | ||||
subtask2 | 0/10 | ||||||
4 | Wrong answer | 39ms | 18012 KiB | ||||
5 | Wrong answer | 45ms | 20352 KiB | ||||
6 | Wrong answer | 50ms | 22504 KiB | ||||
7 | Wrong answer | 52ms | 24936 KiB | ||||
subtask3 | 0/15 | ||||||
8 | Accepted | 48ms | 48024 KiB | ||||
9 | Accepted | 45ms | 48480 KiB | ||||
10 | Wrong answer | 64ms | 50780 KiB | ||||
11 | Wrong answer | 57ms | 50848 KiB | ||||
12 | Wrong answer | 54ms | 50644 KiB | ||||
13 | Wrong answer | 57ms | 52456 KiB | ||||
14 | Wrong answer | 52ms | 51872 KiB | ||||
subtask4 | 0/35 | ||||||
15 | Wrong answer | 3ms | 13832 KiB | ||||
16 | Accepted | 3ms | 13840 KiB | ||||
17 | Accepted | 3ms | 13860 KiB | ||||
18 | Wrong answer | 3ms | 14000 KiB | ||||
19 | Wrong answer | 3ms | 14232 KiB | ||||
20 | Wrong answer | 3ms | 14360 KiB | ||||
21 | Wrong answer | 3ms | 14236 KiB | ||||
22 | Wrong answer | 3ms | 14248 KiB | ||||
subtask5 | 0/40 | ||||||
23 | Wrong answer | 48ms | 31616 KiB | ||||
24 | Accepted | 39ms | 33308 KiB | ||||
25 | Accepted | 37ms | 34848 KiB | ||||
26 | Wrong answer | 52ms | 34148 KiB | ||||
27 | Wrong answer | 48ms | 35120 KiB | ||||
28 | Wrong answer | 54ms | 36972 KiB | ||||
29 | Wrong answer | 50ms | 37920 KiB | ||||
30 | Wrong answer | 52ms | 38460 KiB | ||||
31 | Wrong answer | 50ms | 39212 KiB | ||||
32 | Wrong answer | 54ms | 40128 KiB | ||||
33 | Wrong answer | 57ms | 42104 KiB |