# PBinfo #676 – CountPrimSub – Rezolvare C++

Sursa: Codul lui Andrei
Limbaj: C++
Autor rezolvare: Darius

## Enunț

Cerința
Considerăm un arbore binar cu n noduri în care fiecare nod este numerotat de la 1 la n și conține o valoare număr natural. Se dau k noduri din arbore și se cere determinarea, pentru fiecare nod, a numărului de noduri din subarborele cu rădăcina în acel nod care conțin valori prime.

Date de intrare
Fișierul de intrare countprimsub.in conține pe prima linie numărul n. Fiecare dintre următoarele n linii contine câte 3 numere X st dr; linia i + 1 din fișier conține informatiile despre nodul numerotat cu i: X reprezintă valoare din nod, st reprezintă numărul de ordine al descendentului stâng sau 0 dacă nodul i nu are descendent stâng, iar dr reprezintă numărul de ordine al descendentului drept sau 0 dacă nodul i nu are descendent drept.

Pe următoarea linie se află numărul k, iar pe fiecare dintre următoarele k linii se află câte un număr natural cuprins între 1 și n, reprezentând nodul curent.

Date de ieșire
Fișierul de ieșire countprimsub.out va conține k linii; fiecare linie va conține numărul de noduri din subarborele cu rădăcina în nodul corespunzător.

Restricții și precizări

1 ≤ n ≤ 1000
valorile din nodurile arborelui vor fi mai mici sau egale cu 1.000.000
1 ≤ k ≤ 1000

Exemplu:
countprimsub.in

6
2 3 5
6 0 6
1 0 0
7 1 2
4 0 0
10 0 0
4
1
2
4
2
countprimsub.out

1
0
2
0
Explicație
Exemplul corespunde arborelui de mai jos, în care au fost marcate cu albastru valorile din noduri, iar cu roșu numerele de ordine ale nodurilor.



Subarborele cu rădăcina în nodul numerotat cu 1 conține valorile 1 2 4, dintre care doar 2 este număr prim.

## Cod sursă C++

```cpp
#include <iostream>
#include <fstream>
#define NN 1005
using namespace std;
ifstream fin("countprimsub.in");
ofstream fout("countprimsub.out");
int n, info[NN], st[NN], dr[NN];
void citire()
{
	fin >> n;
	for(int i =1 ; i <= n ; ++i)
		fin >> info[i] >> st[i] >> dr[i];
}
int radacina()
{
	int v[NN];
	for(int i =1 ; i <= n ; ++i)
		v[i] = 0;
	for(int i = 1 ; i <= n ; ++i)
	{
		if(st[i] != 0)
			v[st[i]] = 1;
		if(dr[i] != 0)
			v[dr[i]] = 1;
	}
	for(int i = 1 ; i <= n ; ++i)
		if(v[i] == 0)
			return i;
	return 0;
}
bool prim(int n)
{
	if(n < 2)
		return false;
	if(n > 2 && n % 2 == 0)
		return false;
	for(int d = 3 ; d * d <= n ; d += 2)
		if(n % d == 0)
			return false;
	return true;
}
int count(int x)
{
	if (x)
		if(prim(info[x]))
			return 1 + count(st[x]) + count(dr[x]);
		else
			return count(st[x]) + count(dr[x]);
	else
		return 0;
}
int main()
{
	citire();
	int k , x;
	fin >> k;
	for(int i = 1 ; i <= k ; i ++)
	{
		fin >> x;
		fout << count(x)<< endl;
	}
	return 0;
}
```

## Resurse

- [Pagina originală](https://codulluiandrei.ro/rezolvari/pbinfo/676)
- [Sursa C++](https://codulluiandrei.ro/rezolvari/pbinfo/676/cod.cpp)
- [Sursa pe GitHub](https://raw.githubusercontent.com/codulluiandrei/pbinfo/main/pbinfo-676/main.cpp)
- [Cursuri de informatică pentru BAC](https://codulluiandrei.ro/academia/info-curs/pachet-cpp/)