101822024-03-29 10:45:55111Különböző katicákcpp17Runtime error 0/10067ms45608 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];
	}
	auto vv=v;
	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);
	v[1]=w[1].first;
	auto dfs2=[&](auto self,int i)->void{
		for(int j:g[i]){
			if(j==p[i]){
				continue;
			}
			if(w[j].first==-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++){
		if(v[i]<0)exit(1);
		if(v[i]!=vv[i]&&vv[i]!=-1)exit(1);
		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;
}
SubtaskSumTestVerdictTimeMemory
subtask10/0
1Accepted3ms1704 KiB
2Accepted3ms1924 KiB
3Accepted48ms19464 KiB
subtask20/10
4Runtime error32ms17632 KiB
5Runtime error35ms19300 KiB
6Runtime error39ms20896 KiB
7Runtime error41ms22572 KiB
subtask30/15
8Accepted48ms44396 KiB
9Accepted46ms44104 KiB
10Accepted67ms45608 KiB
11Accepted57ms44672 KiB
12Accepted59ms43760 KiB
13Accepted59ms44816 KiB
14Runtime error48ms43052 KiB
subtask40/35
15Runtime error3ms3696 KiB
16Accepted3ms3900 KiB
17Accepted3ms3936 KiB
18Accepted3ms4020 KiB
19Accepted3ms4276 KiB
20Accepted3ms4168 KiB
21Accepted3ms4420 KiB
22Accepted3ms4564 KiB
subtask50/40
23Runtime error39ms22772 KiB
24Accepted39ms23792 KiB
25Accepted37ms24260 KiB
26Accepted52ms22820 KiB
27Accepted50ms22708 KiB
28Accepted57ms23964 KiB
29Accepted54ms24280 KiB
30Accepted52ms23732 KiB
31Accepted54ms23320 KiB
32Accepted52ms23748 KiB
33Accepted57ms24760 KiB