Submission #15,729

2025-02-24 21:18:55

cpp17

Accepted 100/100

337ms

36784 KiB

// Source: https://usaco.guide/general/io

#include <algorithm>
#include <bits/stdc++.h>
using namespace std;

vector<vector<int>> g , b;
vector<int> a , ans;
vector<array<int , 2>> c;

int t = 0;

void dfs(int h){
    c[h][0] = t++;
    b[a[h]].push_back(c[h][0]);
    for(int x : g[h])dfs(x);
    c[h][1] = t++;
}

int dfs2(int h){
    ans[h] = 1;
    for(int x : g[h])ans[h] = max(dfs2(x) , ans[h]);
    while(*lower_bound(b[ans[h]].begin() , b[ans[h]].end() , c[h][0]) <= c[h][1])ans[h]++;
    return ans[h];
}

int main() {
	int n; cin >> n;
    g.resize(n);
    b.resize(n+2);
    ans.resize(n);
    c.resize(n);
    for(int i = 0; i < n; i++){
        int y; cin >> y;y--;
        if(y != -1)g[y].push_back(i);
    }
    for(int i = 0; i < n; i++){
        int y; cin >> y;
        a.push_back(y);
    }
    dfs(0);
    for(int i = 0; i < b.size(); i++){
        b[i].push_back(1000000);
    }
    dfs2(0);
    for(int x : ans)cout << x << " ";
}

	Sum	Test	Verdict	Time	Memory
subtask1	0/0
		1	Accepted	1ms	316 KiB
		2	Accepted	216ms	24232 KiB
subtask2	5/5
		3	Accepted	1ms	316 KiB
		4	Accepted	1ms	500 KiB
		5	Accepted	1ms	316 KiB
		6	Accepted	1ms	500 KiB
subtask3	8/8
		7	Accepted	1ms	316 KiB
		8	Accepted	1ms	500 KiB
		9	Accepted	1ms	316 KiB
		10	Accepted	1ms	500 KiB
		11	Accepted	158ms	21284 KiB
		12	Accepted	165ms	21220 KiB
		13	Accepted	163ms	20972 KiB
		14	Accepted	192ms	20172 KiB
subtask4	12/12
		15	Accepted	1ms	316 KiB
		16	Accepted	1ms	500 KiB
		17	Accepted	1ms	316 KiB
		18	Accepted	1ms	500 KiB
		19	Accepted	217ms	36784 KiB
		20	Accepted	224ms	36264 KiB
		21	Accepted	238ms	36300 KiB
		22	Accepted	284ms	35132 KiB
		23	Accepted	243ms	36056 KiB
		24	Accepted	233ms	36036 KiB
subtask5	17/17
		25	Accepted	1ms	316 KiB
		26	Accepted	1ms	500 KiB
		27	Accepted	1ms	316 KiB
		28	Accepted	1ms	500 KiB
		29	Accepted	3ms	564 KiB
		30	Accepted	3ms	760 KiB
		31	Accepted	3ms	564 KiB
		32	Accepted	3ms	564 KiB
		33	Accepted	3ms	564 KiB
		34	Accepted	3ms	564 KiB
		35	Accepted	3ms	564 KiB
		36	Accepted	3ms	564 KiB
		37	Accepted	3ms	568 KiB
		38	Accepted	3ms	568 KiB
		39	Accepted	3ms	564 KiB
		40	Accepted	3ms	536 KiB
		41	Accepted	3ms	564 KiB
subtask6	28/28
		42	Accepted	237ms	22824 KiB
		43	Accepted	246ms	24120 KiB
		44	Accepted	237ms	26444 KiB
		45	Accepted	230ms	29648 KiB
		46	Accepted	231ms	32352 KiB
		47	Accepted	238ms	34728 KiB
		48	Accepted	201ms	20264 KiB
		49	Accepted	261ms	35108 KiB
		50	Accepted	239ms	36012 KiB
subtask7	30/30
		51	Accepted	1ms	316 KiB
		52	Accepted	217ms	24232 KiB
		53	Accepted	1ms	316 KiB
		54	Accepted	1ms	500 KiB
		55	Accepted	1ms	316 KiB
		56	Accepted	1ms	500 KiB
		57	Accepted	158ms	21284 KiB
		58	Accepted	165ms	21220 KiB
		59	Accepted	163ms	20972 KiB
		60	Accepted	192ms	20172 KiB
		61	Accepted	217ms	36784 KiB
		62	Accepted	224ms	36264 KiB
		63	Accepted	238ms	36300 KiB
		64	Accepted	284ms	35132 KiB
		65	Accepted	243ms	36056 KiB
		66	Accepted	233ms	36036 KiB
		67	Accepted	3ms	564 KiB
		68	Accepted	3ms	760 KiB
		69	Accepted	3ms	564 KiB
		70	Accepted	3ms	564 KiB
		71	Accepted	3ms	564 KiB
		72	Accepted	3ms	564 KiB
		73	Accepted	3ms	564 KiB
		74	Accepted	3ms	564 KiB
		75	Accepted	3ms	568 KiB
		76	Accepted	3ms	568 KiB
		77	Accepted	3ms	564 KiB
		78	Accepted	3ms	536 KiB
		79	Accepted	3ms	564 KiB
		80	Accepted	237ms	22824 KiB
		81	Accepted	246ms	24120 KiB
		82	Accepted	237ms	26444 KiB
		83	Accepted	230ms	29648 KiB
		84	Accepted	231ms	32352 KiB
		85	Accepted	238ms	34728 KiB
		86	Accepted	201ms	20264 KiB
		87	Accepted	261ms	35108 KiB
		88	Accepted	239ms	36012 KiB
		89	Accepted	239ms	23728 KiB
		90	Accepted	240ms	23732 KiB
		91	Accepted	337ms	22956 KiB
		92	Accepted	300ms	22696 KiB
		93	Accepted	243ms	26844 KiB
		94	Accepted	261ms	29608 KiB
		95	Accepted	261ms	32176 KiB
		96	Accepted	222ms	25312 KiB
		97	Accepted	206ms	25764 KiB
		98	Accepted	207ms	26588 KiB
		99	Accepted	229ms	24232 KiB
		100	Accepted	211ms	25776 KiB
		101	Accepted	212ms	28584 KiB