querymain.cpp
上传用户:qdkongtiao
上传日期:2022-06-29
资源大小:356k
文件大小:3k
源码类别:

书籍源码

开发平台:

Visual C++

  1. /*
  2.  * This file contains code from "C++ Primer, Fourth Edition", by Stanley B.
  3.  * Lippman, Jose Lajoie, and Barbara E. Moo, and is covered under the
  4.  * copyright and warranty notices given in that book:
  5.  * 
  6.  * "Copyright (c) 2005 by Objectwrite, Inc., Jose Lajoie, and Barbara E. Moo."
  7.  * 
  8.  * 
  9.  * "The authors and publisher have taken care in the preparation of this book,
  10.  * but make no expressed or implied warranty of any kind and assume no
  11.  * responsibility for errors or omissions. No liability is assumed for
  12.  * incidental or consequential damages in connection with or arising out of the
  13.  * use of the information or programs contained herein."
  14.  * 
  15.  * Permission is granted for this code to be used for educational purposes in
  16.  * association with the book, given proper citation if and when posted or
  17.  * reproduced.Any commercial use of this code requires the explicit written
  18.  * permission of the publisher, Addison-Wesley Professional, a division of
  19.  * Pearson Education, Inc. Send your request for permission, stating clearly
  20.  * what code you would like to use, and in what specific way, to the following
  21.  * address: 
  22.  * 
  23.  *  Pearson Education, Inc.
  24.  *  Rights and Contracts Department
  25.  *  75 Arlington Street, Suite 300
  26.  *  Boston, MA 02216
  27.  *  Fax: (617) 848-7047
  28. */ 
  29. #include "TextQuery.h"
  30. #include <string>
  31. #include <vector>
  32. #include <map>
  33. #include <set>
  34. #include <iostream>
  35. #include <fstream>
  36. #include <cctype>
  37. #include <cstring>
  38. #include <cstdlib>
  39. using std::set;
  40. using std::string;
  41. using std::map;
  42. using std::vector;
  43. using std::cerr;
  44. using std::cout;
  45. using std::cin;
  46. using std::ifstream;
  47. using std::endl;
  48. string make_plural(size_t, const string&, const string&);
  49. ifstream& open_file(ifstream&, const string&);
  50. void print_results(const set<TextQuery::line_no>& locs, 
  51.                    const string& sought, const TextQuery &file)
  52. {
  53.     // if the word was found, then print count and all occurrences
  54.     typedef set<TextQuery::line_no> line_nums; 
  55.     line_nums::size_type size = locs.size();
  56.     cout << "n" << sought << " occurs "
  57.          << size << " "
  58.          << make_plural(size, "time", "s") << endl;
  59.     // print each line in which the word appeared
  60.     line_nums::const_iterator it = locs.begin();
  61.     for ( ; it != locs.end(); ++it) {
  62.         cout << "t(line "
  63.              // don't confound user with text lines starting at 0
  64.              << (*it) + 1 << ") "
  65.              << file.text_line(*it) << endl;
  66.     }
  67. }
  68. // program takes single argument specifying the file to query
  69. int main(int argc, char **argv)
  70. {
  71.     // open the file from which user will query words
  72.     ifstream infile;
  73.     if (argc < 2 || !open_file(infile, argv[1])) {
  74.         cerr << "No input file!" << endl;
  75.         return EXIT_FAILURE;
  76.     }
  77.     TextQuery tq;
  78.     tq.read_file(infile);  // builds query map
  79.     // iterate with the user: prompt for a word to find and print results
  80.     // loop indefinitely; the loop exit is inside the while
  81.     while (true) {
  82.         cout << "enter word to look for, or q to quit: ";
  83.         string s;
  84.         cin >> s;
  85.         // stop if hit eof on input or a 'q' is entered
  86.         if (!cin || s == "q") break;
  87.         // get the set of line numbers on which this word appears
  88.         set<TextQuery::line_no> locs = tq.run_query(s);
  89.         // print count and all occurrences, if any
  90.         print_results(locs, s, tq);
  91.      }
  92.     return 0;
  93. }