ArraySet.h

Go to the documentation of this file.

#ifndef ARRAYSET_H_
#define ARRAYSET_H_
#include <vector>
#include <algorithm>
using namespace std;
template<typename T>
class ArraySet{
public:
        ArraySet(){
                setsize= 0;
        }
        ArraySet(int maxSize, int (*mapToInt)(T)):indexes(maxSize){
                this->mapToInt = mapToInt;
                fill(indexes.begin(), indexes.end(), -1);
                setsize = 0;
        }
        typedef typename vector<T>::iterator iterator;
        iterator begin(){
                return setElements.begin();
        }
        iterator end(){
                return setElements.end();
        }
        bool insert(T ele){
                int index = (*mapToInt)(ele);
                if(indexes[index]==-1){
                        setsize++;
                        indexes[index] = setElements.size();
                        setElements.push_back(ele);
                        return true;
                }
                return false;
        }
        bool erase(T ele){
                int index = (*mapToInt)(ele);
                if(indexes[index]!=-1){
                        setsize--;
                        int indexEle = indexes[index];
                        //swap(setElements[indexEle], setElements[setElements.size()-1]);
                        setElements[indexEle] = setElements[setElements.size()-1];
                        setElements.pop_back();
                        indexes[(*mapToInt)(setElements[indexEle])] = indexEle;
                        indexes[index] = -1;
                        return true;
                }
                return false;
        }

        void clear(){
                fill(indexes.begin(), indexes.end(), -1);
                setElements.clear();
                setsize = 0;
        }

        int size(){
//              return setElements.size();
                return setsize;
        }

private:
        vector<int> indexes;
        vector<T> setElements;
        int (*mapToInt)(T);
        int setsize;
};

#endif /* ARRAYSET_H_ */

---