Cerința

Se consideră un arbore binar în care nodurile memorează numere naturale nenule și un număr k. În arbore rădăcina se află pe nivelul 0, fii rădăcinii pe nivelul 1, fii fiilor rădăcinii pe nivelul 2, etc. Să se determine suma valorilor din nodurile aflate pe nivelul k.

Date de intrare

Fișierul de intrare knivel1.in conține pe prima linie lista valorilor memorate în nodurile arborelui, obținute în urma parcurgerii în preordine (rădăcină, stâng, drept). Dacă un nod nu are descendent stâng, în listă va apare valoarea 0. Dacă un nod nu are descendent drept, în listă va apare valoarea 0.

Linia a doua a fișierului conține numărul k.

Date de ieșire

Fișierul de ieșire knivel1.out va conține pe prima linie un singur număr S, reprezentând suma cerută.

Restricții și precizări

• se recomandă folosirea arborilor alocați dinamic.

Exemplu

knivel1.in

67 51 18 0 0 24 0 0 48 0 11 0 0
1

knivel1.out

99

Explicație

Exemplul corespunde arborelui desenat mai jos:

Nodurile de pe nivelul 1 sunt 51 48, cu suma 99.

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

ifstream cin("knivel1.in");
ofstream cout("knivel1.out");

int s[1001];

void citire(int nod, int ad){
    int x;
    cin >> x;
    s[ad] += x;
    if(x){
        citire(2 * nod, ad+1);
        citire(2 * nod + 1, ad+1);
    }
}

int main(){
    citire(1, 0);
    int k;
    cin >> k;
    cout << s[k];
    return 0;
}