扩展程序以计算行频率

时间:2011-11-30 06:36:54

标签: c++

我有这个简单的程序来计算单词频率:

#include <iostream>
#include <map>
#include <cctype>
#include <string>
using namespace std;
int main(void) {
  map<string, int> words;
  map<string, int>::iterator i;
string s;
while (cin >> s) {
    transform(s.begin(), s.end(), s.begin(), ::tolower);
words[s]++;
  }
for (i=words.begin(); i != words.end(); i++)
    cout << i->first << " " << i->second << endl;
return 0;

现在我该如何计算行频而不是字频。每行在输入中用'\ n'新行字符分隔。

2 个答案:

答案 0 :(得分:2)

如果你想计算总行数,那么它非常简单:

 size_t countLine = 0;
 while(std::getline(std::cin, s)) countLine++;

但如果您真的想要计算同一行的频率,那么在您的代码中,替换

string s;

line s;

其中line定义为:

struct line : std::string
{
    friend std::istream & operator >> (std::istream &in, line &ln) 
    {
        return std::getline(in, ln);
    }
};

或者,您可以将其设为通用:

template<typename CharT, typename Traits = std::char_traits<CharT>, typename Allocator = std::allocator<CharT>>
struct basic_line : public std::basic_string<CharT, Traits, Allocator>
{
    friend std::basic_istream<CharT, Traits> & operator >> (std::basic_istream<CharT, Traits > &in, basic_line<CharT, Traits, Allocator> &ln) 
    {
        return std::getline(in, ln);
    }
};

然后定义这些typedef:

typedef basic_line<char>    line;
typedef basic_line<wchar_t> wline;

答案 1 :(得分:1)

使用getline(cin,s, '\n')

而不是简单的cin >> s