Advertisement
Mohammad_Dipu_Sultan

Sum of Nodes in Kth Level

Oct 15th, 2023
727
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.76 KB | None | 0 0
  1. #include<bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main(){
  5.     int k;
  6.     cin >> k;
  7.     string s;
  8.     cin >> s;
  9.     int level = -1, sum = 0, n = s.size(), value = 0;
  10.     for(int i = 0; i < n; i++){
  11.         if(s[i] == '(') level++;
  12.         else if(s[i] == ')') level--;
  13.         else if(level == k){
  14.             if(s[i] == '-'){
  15.                 i++;
  16.                 value = 0;
  17.                 while(s[i]!='(' and s[i]!= ')' and s[i]>='0' and s[i]<='9'){
  18.                     value = value * 10 + s[i]-'0';
  19.                     i++;
  20.                 }
  21.                 i--;
  22.                 value = - value;
  23.                 sum+=value;
  24.             }
  25.             else{
  26.                 value = 0;
  27.                 while(s[i]!='(' and s[i]!= ')' and s[i]>='0' and s[i]<='9'){
  28.                     value = value * 10 + s[i]-'0';
  29.                     i++;
  30.                 }
  31.                 i--;
  32.                 sum+=value;
  33.             }
  34.         }
  35.     }
  36.     cout << sum << endl;
  37. }
  38.  
  39. /*
  40. 2
  41. (0(5(16()())(4()(9()())))(7(1()())(3()())))
  42.  
  43. 24
  44. */
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement