Advertisement
Alaricy

Untitled

Oct 12th, 2022
87
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 4.53 KB | None | 0 0
  1. //работает
  2. #include <algorithm>
  3. #include <iostream>
  4. #include <set>
  5. #include <string>
  6. #include <utility>
  7. #include <vector>
  8. #include <map>
  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 = 0;
  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.     int relevance;
  50. };
  51.  
  52. class SearchServer {
  53. public:
  54.     void SetStopWords(const string& text) {
  55.         for (const string& word : SplitIntoWords(text)) {
  56.             stop_words_.insert(word);
  57.         }
  58.     }
  59.  
  60.     void AddDocument(int document_id, const string& document) {
  61.         const vector<string> words = SplitIntoWordsNoStop(document);
  62.         for (string word:words){
  63.            word_to_documents_[word].insert(document_id);
  64.         }        
  65.     }
  66.  
  67.     vector<Document> FindTopDocuments(const string& raw_query) const {
  68.         const QueryContent query_words = ParseQuery(raw_query);
  69.         auto matched_documents = FindAllDocuments(query_words);
  70.  
  71.         sort(matched_documents.begin(), matched_documents.end(),
  72.              [](const Document& lhs, const Document& rhs) {
  73.                  return lhs.relevance > rhs.relevance;
  74.              });
  75.         if (matched_documents.size() > MAX_RESULT_DOCUMENT_COUNT) {
  76.             matched_documents.resize(MAX_RESULT_DOCUMENT_COUNT);
  77.         }
  78.         return matched_documents;
  79.     }
  80.  
  81. private:
  82.     map<string, set<int>>  word_to_documents_;
  83.    
  84.     struct QueryContent {
  85.         set<string> pluswords;
  86.         set<string> minuswords;
  87.     };
  88.  
  89.     set<string> stop_words_;
  90.  
  91.     bool IsStopWord(const string& word) const {
  92.         return stop_words_.count(word) > 0;
  93.     }
  94.  
  95.     vector<string> SplitIntoWordsNoStop(const string& text) const {
  96.         vector<string> words;
  97.         for (const string& word : SplitIntoWords(text)) {
  98.             if (!IsStopWord(word)) {
  99.                 words.push_back(word);
  100.             }
  101.         }
  102.         return words;
  103.     }
  104.  
  105.     QueryContent ParseQuery(const string& text) const {
  106.         QueryContent query_words;
  107.         for (const string& word : SplitIntoWordsNoStop(text)) {
  108.             if (word[0]!='-') query_words.pluswords.insert(word);
  109.             else {
  110.                 query_words.minuswords.insert(word.substr(1));
  111.              }
  112.         }
  113.         return query_words;
  114.     }
  115.  
  116.     vector<Document> FindAllDocuments(const QueryContent query_words) const {
  117.         map <int,int> document_to_relevance;
  118.         vector<Document> matched_documents;
  119.         for (const auto& qwr_wrd : query_words.pluswords)
  120.         {
  121.             if (word_to_documents_.count(qwr_wrd))
  122.             {
  123.                 for (int id:word_to_documents_.at(qwr_wrd))
  124.                 {
  125.                     document_to_relevance[id]++;
  126.                 }
  127.             }
  128.         }
  129.         for (const auto& qwr_mwrd : query_words.minuswords)
  130.         {
  131.             if (word_to_documents_.count(qwr_mwrd))
  132.             {
  133.                 for (int id:word_to_documents_.at(qwr_mwrd))
  134.                 {
  135.                     document_to_relevance.erase(id);
  136.                 }
  137.             }
  138.         }
  139.         for (const auto& dctorlv:document_to_relevance)
  140.         {
  141.            matched_documents.push_back({dctorlv.first,dctorlv.second});    
  142.         }
  143.         return matched_documents;
  144.     }
  145.  
  146.     };
  147.  
  148. SearchServer CreateSearchServer() {
  149.     SearchServer search_server;
  150.     search_server.SetStopWords(ReadLine());
  151.  
  152.     const int document_count = ReadLineWithNumber();
  153.     for (int document_id = 0; document_id < document_count; ++document_id) {
  154.         search_server.AddDocument(document_id, ReadLine());
  155.     }
  156.  
  157.     return search_server;
  158. }
  159.  
  160. int main() {
  161.     const SearchServer search_server = CreateSearchServer();
  162.  
  163.     const string query = ReadLine();
  164.     for (const auto& [document_id, relevance] : search_server.FindTopDocuments(query)) {
  165.         cout << "{ document_id = "s << document_id << ", "
  166.              << "relevance = "s << relevance << " }"s << endl;
  167.     }
  168. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement