A program in both Python and C++ is given below, which performs the following task: read white-space delimited words from stdin, print the unique words sorted by string length along with a count of each unique word to stdout. The format for a line of the output is: length, count, word.
For exmaple, with this input file (488kB of a thesaurus) http://pastebin.com/raw.php?i=NeUBQ22T
The output, with formatting, is this:
1 57 "
1 1 n
1 1 )
1 3 *
1 18 ,
1 7 -
1 1 R
1 13 .
1 2 1
1 1 S
1 5 2
1 1 3
1 2 4
1 2 &
1 91 %
1 1 5
1 1 6
1 1 7
1 1 8
1 2 9
1 16 ;
1 2 =
1 5 A
1 1 C
1 5 e
1 3 E
1 1 G
1 11 I
1 1 L
1 4 N
1 681 a
1 2 y
1 1 P
2 1 67
2 1 y;
2 1 P-
2 85 no
2 9 ne
2 779 of
2 1 n;
...
Here is the program in C++
#include <vector>
#include <string>
#include <iostream>
#include <set>
#include <map>bool compare_strlen (const std::string &lhs, const std::string &rhs) {return (lhs.length() < rhs.length());
}int main (int argc, char *argv[]) {std::string str;std::vector<std::string> words;/* Extract words from the input file, splitting on whitespace */while (std::cin >> str) {words.push_back(str);}/* Extract unique words and count the number of occurances of each word */std::set<std::string> unique_words;std::map<std::string,int> word_count; for (std::vector<std::string>::iterator it = words.begin();it != words.end(); ++it) {unique_words.insert(*it);word_count[*it]++;}words.clear();std::copy(unique_words.begin(), unique_words.end(),std::back_inserter(words));// Sort by word length std::sort(words.begin(), words.end(), compare_strlen);// Print words with length and number of occurancesfor (std::vector<std::string>::iterator it = words.begin();it != words.end(); ++it) {std::cout << it->length() << " " << word_count[*it] << " " <<*it << std::endl;}return 0;
}
Here is the Program in Python:
import fileinput
from collections import defaultdictwords = set()
count = {}
for line in fileinput.input():line_words = line.split()for word in line_words:if word not in words:words.add(word)count[word] = 1else:count[word] += 1 words = list(words)
words.sort(key=len)for word in words:print len(word), count[word], word
For the C++ program, the compiler used was g++ 4.9.0 with the -O3 flag.
The version of Python used was 2.7.3
Time taken for the C++ program:
time ./main > measure-and-count.txt < ~/Documents/thesaurus/thesaurus.txtreal 0m0.687s
user 0m0.559s
sys 0m0.123s
Time taken for the Python program:
time python main.py > measure-and-count.txt < ~/Documents/thesaurus/thesaurus.txtreal 0m0.369s
user 0m0.308s
sys 0m0.029s
The Python program is much faster than the C++ program, and is relatively even faster with larger input sizes. What's going on here? Is my use of the C++ STL incorrect?
Edit:
As suggested by a comment and an answer, I changed the C++ program to use std::unordered_set
and std::unordered_map
.
The following lines were changed
#include <unordered_set>
#include <unordered_map>...std::unordered_set<std::string> unique_words;
std::unordered_map<std::string,int> word_count;
The compilation command:
g++-4.9 -std=c++11 -O3 -o main main.cpp
This improved performance only marginally:
time ./main > measure-and-count.txt < ~/Documents/thesaurus/thesaurus.txtreal 0m0.604s
user 0m0.479s
sys 0m0.122s
Edit2: A much faster program in C++
This is a combination of NetVipeC's solution, Dieter Lücking's solution, and the top answer to this question. The real performance killer was cin
using an unbuffered read by default. Solved with std::cin.sync_with_stdio(false);
. This solution also uses a single container, taking advantage of the ordered map
in C++.
#include <vector>
#include <string>
#include <iostream>
#include <set>
#include <map>struct comparer_strlen {bool operator()(const std::string& lhs, const std::string& rhs) const {if (lhs.length() == rhs.length())return lhs < rhs;return lhs.length() < rhs.length();}
};int main(int argc, char* argv[]) {std::cin.sync_with_stdio(false);std::string str;typedef std::map<std::string, int, comparer_strlen> word_count_t;/* Extract words from the input file, splitting on whitespace *//* Extract unique words and count the number of occurances of each word */word_count_t word_count;while (std::cin >> str) {word_count[str]++;}// Print words with length and number of occurancesfor (word_count_t::iterator it = word_count.begin();it != word_count.end(); ++it) {std::cout << it->first.length() << " " << it->second << " "<< it->first << '\n';}return 0;
}
Runtime
time ./main3 > measure-and-count.txt < ~/Documents/thesaurus/thesaurus.txtreal 0m0.106s
user 0m0.091s
sys 0m0.012s
Edit3: A nice and concise version of the Python program was provided by Daniel, it runs in about the same time as the version above:
import fileinput
from collections import Countercount = Counter(w for line in fileinput.input() for w in line.split())for word in sorted(count, key=len):print len(word), count[word], word
Runtime:
time python main2.py > measure-and-count.txt.py < ~/Documents/thesaurus/thesaurus.txtreal 0m0.342s
user 0m0.312s
sys 0m0.027s