dictionnaire_arbre.hpp

Go to the documentation of this file.

#ifndef DICTIONNAIRE_HPP
#define DICTIONNAIRE_HPP

#include "treestring.hpp"
#include <utility>

bool triPair(const pair<string, int> &first, const pair<string, int> &second){
    return first.second > second.second;
}

class Dictionnaire{
    
    private :
        TreeString dico;

    public :
    
        Dictionnaire(){}
        
        ~Dictionnaire(){}
    
        bool contientMot(string mot){
            bool trouve=false;
            forward_list<string> list;
            dico.getWords(list);
            auto it = list.begin();
            while(it!=list.end() and !trouve){
                trouve = (it->compare(mot)==0);
                ++it;
            }
            return trouve;
            
        }           
        
        void ajouterMot(string mot){
            dico.put(mot);
        }   

        void associerMot(string mot){
            dico.put(mot);
        }   
            
        int valeurAssociee(string mot){
            bool trouve=false;
            forward_list<pair<string,int>> words;
            dico.getWordsFrequencies(words);
            auto it = words.begin();
            while(it!=words.end() and !trouve){
                trouve = ((it->first).compare(mot)==0);
                ++it;
            }
            return it->second;
        }
                
        void plusFrequentes(pair<string,int> *frequences){
            forward_list<pair<string, int>> pairs;
            // on récupère les mots avec leurs fréquences
            dico.getWordsFrequencies(pairs);
            // on tri la liste
            pairs.sort(&triPair); // complexité : n(log(n))
            auto it_pairs = pairs.begin();
            int i = 0;
            // on met les dix premières pairs dans le tableau
            while(i<10 and pairs.end() != it_pairs){
                frequences[i++] = *it_pairs;
                ++it_pairs;
            }
        }
};

#endif // DICTIONNAIRE_HPP