oj.uz

Submission #924903

# Submission time Handle Problem Language Result Execution time Memory
924903 2024-02-10T03:41:57 Z Faisal_Saqib Cat in a tree (BOI17_catinatree) C++17
0 / 100
 3 ms 10428 KB 
#include <iostream>
#include <vector>
#include <bitset>
using namespace std;
const int N=2e5+10;
int n,d;
vector<int> ma[N],vep[N];
int h[N];
bitset<N> rem;
void remove(int x,int dp=0,int p=-1)
{
	if(rem[x])
		return;
	rem.set(x);
	if(dp==(d-1))
		return;
	for(int y:ma[x])
		if(y!=p)
			remove(y,dp+1,x);
}
int main()
{
	cin.tie(0);
	cout.tie(0);
	ios::sync_with_stdio(0);
	cin>>n>>d;
	if(d==1)
	{
		cout<<n<<endl;
		return 0;
	}
	vep[0].push_back(1);
	int max_h=0;
	for(int i=2;i<=n;i++)
	{
		int p;
		cin>>p;
		p++;
		h[i]=h[p]+1;
		vep[h[i]].push_back(i);
		max_h=max(max_h,h[i]);
		ma[p].push_back(i);
		ma[i].push_back(p);
	}
	int ans=0;
	for(int cur_h=max_h;cur_h>=0;cur_h--)
	{
		for(int v:vep[cur_h])
		{
			if(!rem[v])
			{
				ans++;
				remove(v);
			}
		}
	}
	cout<<ans<<'\n';
	return 0;
}

Subtask #1
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 10328 KB Output is correct
2 Correct 3 ms 10332 KB Output is correct
3 Correct 3 ms 10332 KB Output is correct
4 Incorrect 3 ms 10428 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #2
0 / 40.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 10328 KB Output is correct
2 Correct 3 ms 10332 KB Output is correct
3 Correct 3 ms 10332 KB Output is correct
4 Incorrect 3 ms 10428 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #3
0 / 49.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 10328 KB Output is correct
2 Correct 3 ms 10332 KB Output is correct
3 Correct 3 ms 10332 KB Output is correct
4 Incorrect 3 ms 10428 KB Output isn't correct
5 Halted 0 ms 0 KB -