zimovid

Untitled

Jun 16th, 2023
732
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 8.44 KB | None | 0 0
  1. #include <algorithm>
  2. #include <cmath>
  3. #include <iostream>
  4. #include <map>
  5. #include <set>
  6. #include <string>
  7. #include <utility>
  8. #include <vector>
  9.  
  10. using namespace std;
  11.  
  12. const int MAX_RESULT_DOCUMENT_COUNT = 5;
  13.  
  14. string ReadLine() {
  15.     string s;
  16.     getline(cin, s);
  17.     return s;
  18. }
  19.  
  20. int ReadLineWithNumber() {
  21.     int result;
  22.     cin >> result;
  23.     ReadLine();
  24.     return result;
  25. }
  26.  
  27. vector<string> SplitIntoWords(const string& text) {
  28.     vector<string> words;
  29.     string word;
  30.     for (const char c : text) {
  31.         if (c == ' ') {
  32.             if (!word.empty()) {
  33.                 words.push_back(word);
  34.                 word.clear();
  35.             }
  36.         } else {
  37.             word += c;
  38.         }
  39.     }
  40.     if (!word.empty()) {
  41.         words.push_back(word);
  42.     }
  43.  
  44.     return words;
  45. }
  46.  
  47. struct Document {
  48.     int id;
  49.     double relevance;
  50.     int rating;
  51. };
  52.  
  53. enum class DocumentStatus {
  54.     ACTUAL,
  55.     IRRELEVANT,
  56.     BANNED,
  57.     REMOVED,
  58. };
  59.  
  60. class SearchServer {
  61. public:
  62.     int GetDocumentCount() const {
  63.         return static_cast<int>(documents_.size());
  64.     }
  65.    
  66.     tuple<vector<string>, DocumentStatus> MatchDocument(const string& raw_query, int document_id) const {
  67.         const Query query = ParseQuery(raw_query);
  68.  
  69.         // Check if the document matches the query.
  70.         bool is_match = true;
  71.         if (!query.minus_words.empty()) {  // Check only when there are minus-words.
  72.             for (const string& word : query.minus_words) {
  73.                 if (word_to_document_freqs_.count(word) > 0 &&
  74.                     word_to_document_freqs_.at(word).count(document_id) > 0) {
  75.                     is_match = false;  // The document contains a minus-word.
  76.                     break;
  77.                 }
  78.             }
  79.         }
  80.  
  81.         if (is_match) {
  82.             vector<string> matched_words;
  83.             for (const string& word : query.plus_words) {
  84.                 if (word_to_document_freqs_.count(word) > 0 &&
  85.                     word_to_document_freqs_.at(word).count(document_id) > 0) {
  86.                     matched_words.push_back(word);
  87.                 }
  88.             }
  89.             sort(matched_words.begin(), matched_words.end());
  90.             matched_words.erase(unique(matched_words.begin(), matched_words.end()), matched_words.end());
  91.  
  92.             return {matched_words, documents_.at(document_id).status};
  93.         } else {
  94.             return {{}, DocumentStatus::IRRELEVANT};
  95.         }
  96.     }
  97.    
  98.     void SetStopWords(const string& text) {
  99.         for (const string& word : SplitIntoWords(text)) {
  100.             stop_words_.insert(word);
  101.         }
  102.     }
  103.  
  104.     void AddDocument(int document_id, const string& document, DocumentStatus status,
  105.                      const vector<int>& ratings) {
  106.         const vector<string> words = SplitIntoWordsNoStop(document);
  107.         const double inv_word_count = 1.0 / words.size();
  108.         for (const string& word : words) {
  109.             word_to_document_freqs_[word][document_id] += inv_word_count;
  110.         }
  111.         documents_.emplace(document_id, DocumentData{ComputeAverageRating(ratings), status});
  112.     }
  113.  
  114.     vector<Document> FindTopDocuments(const string& raw_query,
  115.                                       DocumentStatus status = DocumentStatus::ACTUAL) const {
  116.         const Query query = ParseQuery(raw_query);
  117.         auto matched_documents = FindAllDocuments(query, status);
  118.  
  119.         sort(matched_documents.begin(), matched_documents.end(),
  120.              [](const Document& lhs, const Document& rhs) {
  121.                  return lhs.relevance > rhs.relevance;
  122.              });
  123.         if (matched_documents.size() > MAX_RESULT_DOCUMENT_COUNT) {
  124.             matched_documents.resize(MAX_RESULT_DOCUMENT_COUNT);
  125.         }
  126.         return matched_documents;
  127.     }
  128.  
  129. private:
  130.     struct DocumentData {
  131.         int rating;
  132.         DocumentStatus status;
  133.     };
  134.  
  135.     set<string> stop_words_;
  136.     map<string, map<int, double>> word_to_document_freqs_;
  137.     map<int, DocumentData> documents_;
  138.  
  139.     bool IsStopWord(const string& word) const {
  140.         return stop_words_.count(word) > 0;
  141.     }
  142.  
  143.     vector<string> SplitIntoWordsNoStop(const string& text) const {
  144.         vector<string> words;
  145.         for (const string& word : SplitIntoWords(text)) {
  146.             if (!IsStopWord(word)) {
  147.                 words.push_back(word);
  148.             }
  149.         }
  150.         return words;
  151.     }
  152.  
  153.     static int ComputeAverageRating(const vector<int>& ratings) {
  154.         if (ratings.empty()) {
  155.             return 0;
  156.         }
  157.         int rating_sum = 0;
  158.         for (const int rating : ratings) {
  159.             rating_sum += rating;
  160.         }
  161.         return rating_sum / static_cast<int>(ratings.size());
  162.     }
  163.  
  164.     struct QueryWord {
  165.         string data;
  166.         bool is_minus;
  167.         bool is_stop;
  168.     };
  169.  
  170.     QueryWord ParseQueryWord(string text) const {
  171.         bool is_minus = false;
  172.         // Word shouldn't be empty
  173.         if (text[0] == '-') {
  174.             is_minus = true;
  175.             text = text.substr(1);
  176.         }
  177.         return {text, is_minus, IsStopWord(text)};
  178.     }
  179.  
  180.     struct Query {
  181.         set<string> plus_words;
  182.         set<string> minus_words;
  183.     };
  184.  
  185.     Query ParseQuery(const string& text) const {
  186.         Query query;
  187.         for (const string& word : SplitIntoWords(text)) {
  188.             const QueryWord query_word = ParseQueryWord(word);
  189.             if (!query_word.is_stop) {
  190.                 if (query_word.is_minus) {
  191.                     query.minus_words.insert(query_word.data);
  192.                 } else {
  193.                     query.plus_words.insert(query_word.data);
  194.                 }
  195.             }
  196.         }
  197.         return query;
  198.     }
  199.  
  200.     // Existence required
  201.     double ComputeWordInverseDocumentFreq(const string& word) const {
  202.         return log(documents_.size() * 1.0 / word_to_document_freqs_.at(word).size());
  203.     }
  204.  
  205.     vector<Document> FindAllDocuments(const Query& query, DocumentStatus status) const {
  206.         map<int, double> document_to_relevance;
  207.         for (const string& word : query.plus_words) {
  208.             if (word_to_document_freqs_.count(word) == 0) {
  209.                 continue;
  210.             }
  211.             const double inverse_document_freq = ComputeWordInverseDocumentFreq(word);
  212.             for (const auto [document_id, term_freq] : word_to_document_freqs_.at(word)) {
  213.                 if (documents_.at(document_id).status == status) {
  214.                     document_to_relevance[document_id] += term_freq * inverse_document_freq;
  215.                 }
  216.             }
  217.         }
  218.  
  219.         for (const string& word : query.minus_words) {
  220.             if (word_to_document_freqs_.count(word) == 0) {
  221.                 continue;
  222.             }
  223.             for (const auto [document_id, _] : word_to_document_freqs_.at(word)) {
  224.                 document_to_relevance.erase(document_id);
  225.             }
  226.         }
  227.  
  228.         vector<Document> matched_documents;
  229.         for (const auto [document_id, relevance] : document_to_relevance) {
  230.             matched_documents.push_back(
  231.                 {document_id, relevance, documents_.at(document_id).rating});
  232.         }
  233.         return matched_documents;
  234.     }
  235. };
  236.  
  237. void PrintMatchDocumentResult(int document_id, const vector<string>& words, DocumentStatus status) {
  238.     cout << "{ "s
  239.          << "document_id = "s << document_id << ", "s
  240.          << "status = "s << static_cast<int>(status) << ", "s
  241.          << "words ="s;
  242.     for (const string& word : words) {
  243.         cout << ' ' << word;
  244.     }
  245.     cout << "}"s << endl;
  246. }
  247. int main() {
  248.     SearchServer search_server;
  249.     search_server.SetStopWords("и в на"s);
  250.     search_server.AddDocument(0, "белый кот и модный ошейник"s,        DocumentStatus::ACTUAL, {8, -3});
  251.     search_server.AddDocument(1, "пушистый кот пушистый хвост"s,       DocumentStatus::ACTUAL, {7, 2, 7});
  252.     search_server.AddDocument(2, "ухоженный пёс выразительные глаза"s, DocumentStatus::ACTUAL, {5, -12, 2, 1});
  253.     search_server.AddDocument(3, "ухоженный скворец евгений"s,         DocumentStatus::BANNED, {9});
  254.     const int document_count = search_server.GetDocumentCount();
  255.     for (int document_id = 0; document_id < document_count; ++document_id) {
  256.         const auto [words, status] = search_server.MatchDocument("пушистый кот"s, document_id);
  257.         PrintMatchDocumentResult(document_id, words, status);
  258.     }
  259. }
  260.  
Advertisement
Add Comment
Please, Sign In to add comment