Advertisement
Kali_prasad

sum of root to path numbers

May 11th, 2022
40
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.75 KB | None | 0 0
  1. #pragma GCC optimize ("O3")
  2. #pragma GCC target ("sse4")
  3.  
  4. #include <bits/stdc++.h>
  5. using namespace std;
  6.  
  7. typedef long long ll;
  8. typedef pair<int, int> pii;
  9. typedef pair<string,int> psi;
  10. typedef unordered_map<int,int> mii;
  11. typedef unordered_map<long long,long long> mll;
  12. typedef unordered_map<string,int> msi;
  13. typedef unordered_map<char,int> mci;
  14. typedef unordered_set<int> si;
  15. typedef unordered_set<long long> sll;
  16. typedef unordered_set<string> ss;
  17. typedef unordered_set<char> sc;
  18. typedef map<int,int> ormii;
  19. typedef map<long long,long long> ormll;
  20. typedef map<string,int> ormsi;
  21. typedef map<char,int> ormci;
  22. typedef set<int> orsi;
  23. typedef set<long long> orsll;
  24. typedef set<string> orss;
  25. typedef set<char> orsc;
  26. typedef vector<int> vi;
  27. typedef vector<string> vs;
  28. typedef vector<char> vc;
  29. typedef vector<ll> vll;
  30. typedef vector<vector<int>> vvi;
  31. typedef vector<vector<string>> vvs;
  32. typedef vector<vector<ll>> vvll;
  33.  
  34. #define FOR(i, a, b) for (auto i=a; i<=(b); i++)
  35. #define FORd(i,b,a) for (int i =b; i >= a; i--)
  36. #define sortinc(v)  sort(v.begin(),v.end())
  37. #define sortdec(v)  sort(v.rbegin(),v.rend())
  38. #define sz(x) (int)(x).size()
  39. #define mp make_pair
  40. #define pb push_back
  41. #define pob pop_back
  42. #define pf push_front
  43. #define pof pop_front
  44. #define fi first
  45. #define se second
  46. #define ins insert
  47.  
  48. const int MOD = 1000000007;
  49. //type functions here
  50.  
  51. class Node
  52. {
  53.     public:
  54.   Node* left;
  55.   Node* right;
  56.     int data;
  57.     Node(int data)
  58.     {
  59.        this->data=data;
  60.         left=NULL;
  61.         right=NULL;
  62.     }
  63. };
  64. int sum;
  65. Node* insert(Node* root,int data)
  66. {
  67.     if(!root) return new Node(data);
  68.     if(data<root->data)
  69.         root->left=insert(root->left,data);
  70.     else
  71.         root->right=insert(root->right,data);
  72.     return root;        
  73. }
  74.  
  75. int digits(int d)
  76. {
  77.     if(d==0) return 1;
  78.     int c=0;
  79.     while(d)
  80.     {
  81.         d/=10;
  82.         c++;
  83.     }
  84.     return c;
  85. }
  86. void print(Node* root,int temp)
  87. {
  88.     if(root==NULL) return;
  89.     int sizee=digits(root->data);
  90.        
  91.     while(sizee){
  92.     temp=(temp*10)%MOD;
  93.     sizee--;
  94.     }
  95.    
  96.     temp=(temp+root->data)%MOD;
  97.     if(!root->left&&!root->right)
  98.     {
  99.         sum=(sum+temp)%MOD;
  100.     }
  101.     if(root->left)print(root->left,temp);
  102.     if(root->right)print(root->right,temp);
  103.    
  104.    
  105. }
  106.  
  107. int main() {
  108.     ios_base::sync_with_stdio(false);
  109.     cin.tie(NULL);
  110.    
  111.     int tc=1;
  112.     cin>>tc;
  113.     FOR(w,1,tc)
  114.     {
  115.        int n;
  116.         cin>>n;
  117.         Node* root=NULL;
  118.         FOR(ww,1,n)
  119.         {
  120.             int temp;
  121.             cin>>temp;
  122.             root=insert(root,temp);
  123.         }
  124.         if(root==NULL) continue;
  125.         sum=0;
  126.         print(root,0);
  127.         cout<<sum<<endl;
  128.        
  129.     }
  130.     return 0;
  131. }
  132.  
  133.  
  134.  
  135.  
  136.  
  137.  
  138.  
  139.  
  140.  
  141.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement