時間があったので、セットとリストでやってみました。リストは、最後に挿入された n 個の要素を追跡するだけです。また、一般的な消去を実装しました。パフォーマンスを向上させるには (set が順序付けられているという事実を利用していない場合)、unordered_set の使用を検討してください。include <set>
(とinclude <unordered_set>
同様にstd::set
置き換えますstd::unordered_set
)
#include <set>
#include <list>
#include <assert.h>
template<typename T>
struct setkeepn {
std::set<T> mSet;
std::list<T> mLast;
void insert(T element) {
if (mSet.find(element) == mSet.end()) {
assert(mSet.size() == mLast.size());
// put your limit of 30 below
if (mSet.size() >= 2) {
T extra = mLast.back();
mSet.erase(extra);
mLast.pop_back();
}
mSet.insert(element);
mLast.push_front(element);
}
}
void erase(T element)
{
typename std::set<T>::iterator lToEraseFromSet = mSet.find(element);
if (lToEraseFromSet != mSet.end()) {
// linear on the number of elements.
typename std::list<T>::iterator lToEraseFromList = std::find(mLast.begin(),mLast.end(), element);
assert(lToEraseFromList != mLast.end());
mSet.erase(lToEraseFromSet);
mLast.erase(lToEraseFromList);
}
}
};
int main(int argc, const char * argv[]) {
setkeepn<int> lTest;
lTest.insert(1);
lTest.insert(2);
lTest.insert(2);
lTest.insert(3);
printf("should see 2 3\n");
for (auto e : lTest.mSet) { // 2,3
printf("elements: %d\n",e);
}
lTest.insert(4);
lTest.erase(3);
printf("should see 4\n");
for (auto e : lTest.mSet) { // 4
printf("elements: %d\n",e);
}
return 0;
}
結果:
should see 2 3
elements: 2
elements: 3
should see 4
elements: 4