Submission #1825441


Source Code Expand

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace std;
using namespace __gnu_pbds;

#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define fbo find_by_order
#define ook order_of_key

typedef long long ll;
typedef pair<int,int> ii;
typedef vector<int> vi;
typedef long double ld; 
typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> pbds;
typedef set<int>::iterator sit;
typedef map<int,int>::iterator mit;
typedef vector<int>::iterator vit;

int a[100001];
vi adj[100001];
int h[100001];
int ans, k;

void dfs(int u, int p)
{
	h[u] = 1;
	for(int i = 0; i < adj[u].size(); i++)
	{
		int v = adj[u][i];
		dfs(v, u);
		h[u] = max(h[u], h[v] + 1);
	}
	if(u != 1 && p != 1 && h[u] >= k)
	{
		ans++; h[u] = 0;
	}
}

int main()
{
	ios_base::sync_with_stdio(0); cin.tie(0);
	int n; cin >> n >> k;
	for(int i = 1; i <= n; i++)
	{
		cin >> a[i];
	}
	ans = 0;
	if(a[1] != 1)
	{
		a[1] = 1;
		ans++;
	}
	for(int i = 2; i <= n; i++)
	{
		adj[a[i]].pb(i);
	}
	//now we have a tree
	dfs(1, -1);
	cout << ans;
}	

Submission Info

Submission Time
Task D - Teleporter
User vjudge4
Language C++14 (GCC 5.4.1)
Score 800
Code Size 1132 Byte
Status AC
Exec Time 29 ms
Memory 14336 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 800 / 800
Status
AC × 3
AC × 58
Set Name Test Cases
Sample 0_00.txt, 0_01.txt, 0_02.txt
All 0_00.txt, 0_01.txt, 0_02.txt, 1_00.txt, 1_01.txt, 1_02.txt, 1_03.txt, 1_04.txt, 1_05.txt, 1_06.txt, 1_07.txt, 1_08.txt, 1_09.txt, 1_10.txt, 1_11.txt, 1_12.txt, 1_13.txt, 1_14.txt, 1_15.txt, 1_16.txt, 1_17.txt, 1_18.txt, 1_19.txt, 1_20.txt, 1_21.txt, 1_22.txt, 1_23.txt, 1_24.txt, 1_25.txt, 1_26.txt, 1_27.txt, 1_28.txt, 1_29.txt, 1_30.txt, 1_31.txt, 1_32.txt, 1_33.txt, 1_34.txt, 1_35.txt, 1_36.txt, 1_37.txt, 1_38.txt, 1_39.txt, 1_40.txt, 1_41.txt, 1_42.txt, 1_43.txt, 1_44.txt, 1_45.txt, 1_46.txt, 1_47.txt, 1_48.txt, 1_49.txt, 1_50.txt, 1_51.txt, 1_52.txt, 1_53.txt, 1_54.txt
Case Name Status Exec Time Memory
0_00.txt AC 3 ms 2684 KB
0_01.txt AC 2 ms 2560 KB
0_02.txt AC 2 ms 2560 KB
1_00.txt AC 2 ms 2560 KB
1_01.txt AC 2 ms 2560 KB
1_02.txt AC 28 ms 14336 KB
1_03.txt AC 28 ms 14336 KB
1_04.txt AC 28 ms 14336 KB
1_05.txt AC 28 ms 14336 KB
1_06.txt AC 28 ms 14336 KB
1_07.txt AC 29 ms 14336 KB
1_08.txt AC 29 ms 14336 KB
1_09.txt AC 28 ms 14336 KB
1_10.txt AC 26 ms 10368 KB
1_11.txt AC 26 ms 10368 KB
1_12.txt AC 26 ms 10368 KB
1_13.txt AC 24 ms 6528 KB
1_14.txt AC 23 ms 6528 KB
1_15.txt AC 23 ms 6528 KB
1_16.txt AC 16 ms 5116 KB
1_17.txt AC 16 ms 5116 KB
1_18.txt AC 16 ms 5116 KB
1_19.txt AC 10 ms 3960 KB
1_20.txt AC 10 ms 3960 KB
1_21.txt AC 10 ms 3960 KB
1_22.txt AC 21 ms 4992 KB
1_23.txt AC 21 ms 4992 KB
1_24.txt AC 21 ms 4992 KB
1_25.txt AC 20 ms 4480 KB
1_26.txt AC 20 ms 4480 KB
1_27.txt AC 20 ms 4480 KB
1_28.txt AC 13 ms 4096 KB
1_29.txt AC 13 ms 4096 KB
1_30.txt AC 13 ms 4096 KB
1_31.txt AC 22 ms 4992 KB
1_32.txt AC 22 ms 4992 KB
1_33.txt AC 22 ms 4992 KB
1_34.txt AC 22 ms 4992 KB
1_35.txt AC 23 ms 5376 KB
1_36.txt AC 23 ms 5376 KB
1_37.txt AC 23 ms 5376 KB
1_38.txt AC 23 ms 5376 KB
1_39.txt AC 23 ms 5376 KB
1_40.txt AC 23 ms 5376 KB
1_41.txt AC 23 ms 5376 KB
1_42.txt AC 23 ms 5376 KB
1_43.txt AC 23 ms 5760 KB
1_44.txt AC 23 ms 5760 KB
1_45.txt AC 23 ms 5760 KB
1_46.txt AC 23 ms 5760 KB
1_47.txt AC 24 ms 7680 KB
1_48.txt AC 25 ms 7680 KB
1_49.txt AC 25 ms 7680 KB
1_50.txt AC 25 ms 7680 KB
1_51.txt AC 27 ms 12800 KB
1_52.txt AC 27 ms 12800 KB
1_53.txt AC 27 ms 12800 KB
1_54.txt AC 27 ms 12800 KB