Advertisement
Guest User

Untitled

a guest
Nov 17th, 2019
91
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.92 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4.  
  5. /*
  6.  * Complete the contacts function below.
  7.  */
  8.  
  9. struct TrieNode{
  10.     std::vector<int> val;
  11.     std::vector<TrieNode*> children;
  12.     TrieNode(int f) {
  13.         children = std::vector<TrieNode*> (26,nullptr);
  14.         val = std::vector<int> (26,f);
  15.     }
  16. };
  17.  
  18. void Push(std::string& s, int ii, TrieNode* root) {
  19.     if (ii==s.length()) {
  20.         return;
  21.     }
  22.     if (root->children[s[ii]-'a']) {
  23.         root->val[s[ii]-'a'] += 1;
  24.         Push (s, ii+1, root->children[s[ii]-'a']);
  25.     }
  26.     else {
  27.         root->children[s[ii]-'a'] = new TrieNode(0);
  28.         root->val[s[ii]-'a'] += 1;
  29.         Push (s, ii+1, root->children[s[ii]-'a']);
  30.     }
  31. }
  32.  
  33.  
  34. vector<int> contacts(vector<vector<string>> queries) {
  35.     /*
  36.      * Write your code here.
  37.      */
  38.     TrieNode* root = new TrieNode (0);
  39.     vector<int> out;
  40.     for (const auto& it: queries) {
  41.         if (it[0]=="add") {
  42.             Push(it[1], 0, root);
  43.         }
  44.         else {
  45.            
  46.         }
  47.     }
  48.  
  49.  
  50. }
  51.  
  52. int main()
  53. {
  54.     ofstream fout(getenv("OUTPUT_PATH"));
  55.  
  56.     int queries_rows;
  57.     cin >> queries_rows;
  58.     cin.ignore(numeric_limits<streamsize>::max(), '\n');
  59.  
  60.     vector<vector<string>> queries(queries_rows);
  61.     for (int queries_row_itr = 0; queries_row_itr < queries_rows; queries_row_itr++) {
  62.         queries[queries_row_itr].resize(2);
  63.  
  64.         for (int queries_column_itr = 0; queries_column_itr < 2; queries_column_itr++) {
  65.             cin >> queries[queries_row_itr][queries_column_itr];
  66.         }
  67.  
  68.         cin.ignore(numeric_limits<streamsize>::max(), '\n');
  69.     }
  70.  
  71.     vector<int> result = contacts(queries);
  72.  
  73.     for (int result_itr = 0; result_itr < result.size(); result_itr++) {
  74.         fout << result[result_itr];
  75.  
  76.         if (result_itr != result.size() - 1) {
  77.             fout << "\n";
  78.         }
  79.     }
  80.  
  81.     fout << "\n";
  82.  
  83.     fout.close();
  84.  
  85.     return 0;
  86. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement