Upload New File

2022-11-13 11:00:17 +00:00
parent 4899edff93
commit 9747f39c56
1 changed files with 98 additions and 0 deletions
--- a/HeavyKeeper/heavykeeper.cpp
+++ b/HeavyKeeper/heavykeeper.cpp
@@ -0,0 +1,98 @@
+#include <cmath>
+#include <cstdio>
+#include <cstdlib>
+#include <iostream>
+#include <algorithm>
+#include <string>
+#include <cstring>
+#include "../init/BOBHash32.h"
+#include "../init/BOBHash64.h"
+#include "../init/params.h"
+#include "../init/ssummary.h"
+#include "../include/heavykeeper.h"
+using namespace std;
+        heavykeeper::heavykeeper(int M2,int K):M2(M2),K(K) {ss=new ssummary(K); ss->clear(); bobhash=new BOBHash64(1005);}
+        void heavykeeper::clear()
+        {
+            for (int i=0; i<HK_d; i++)
+                for (int j=0; j<=M2+5; j++) HK[i][j].C=HK[i][j].FP=0;
+        }
+        unsigned long long heavykeeper::Hash(string ST)
+        {
+            return (bobhash->run(ST.c_str(),ST.size()));
+        }
+        void heavykeeper::Insert(string x)
+        {
+            bool mon=false;
+            int p=ss->find(x);
+            if (p) mon=true;
+            int maxv=0;
+            unsigned long long H=Hash(x); int FP=(H>>48);
+            for (int j=0; j<HK_d; j++)
+            {
+                int Hsh=H%(M2-(2*HK_d)+2*j+3);
+                int c=HK[j][Hsh].C;
+                if (HK[j][Hsh].FP==FP)
+                {
+                    if (mon || c<=ss->getmin())
+                      HK[j][Hsh].C++;
+                    maxv=max(maxv,HK[j][Hsh].C);
+                } else
+                {
+                    if (!(rand()%int(pow(HK_b,HK[j][Hsh].C))))
+                    {
+                        HK[j][Hsh].C--;
+                        if (HK[j][Hsh].C<=0)
+                        {
+                            HK[j][Hsh].FP=FP;
+                            HK[j][Hsh].C=1;
+                            maxv=max(maxv,1);
+                        }
+                    }
+                }
+            }
+            if (!mon)
+            {
+                if (maxv-(ss->getmin())==1 || ss->tot<K)
+                {
+                    int i=ss->getid();
+                    ss->add2(ss->location(x),i);
+                    ss->str[i]=x;
+                    ss->sum[i]=maxv;
+                    ss->link(i,0);
+                    while(ss->tot>K)
+                    {
+                        int t=ss->Right[0];
+                        int tmp=ss->head[t];
+                        ss->cut(ss->head[t]);
+                        ss->recycling(tmp);
+                    }
+                }
+            } else
+            if (maxv>ss->sum[p])
+            {
+                int tmp=ss->Left[ss->sum[p]];
+                ss->cut(p);
+                if(ss->head[ss->sum[p]]) tmp=ss->sum[p];
+                ss->sum[p]=maxv;
+                ss->link(p,tmp);
+            }
+        }
+        
+        void heavykeeper::work()
+        {
+            int CNT=0;
+            for(int i=N;i;i=ss->Left[i])
+                for(int j=ss->head[i];j;j=ss->Next[j]) {q[CNT].x=ss->str[j]; q[CNT].y=ss->sum[j]; CNT++; }
+            sort(q,q+CNT,cmp);
+        }
+        pair<string ,int> heavykeeper::Query(int k)
+        {
+            return make_pair(q[k].x,q[k].y);
+        }
+
+
+       // heavykeeper::~heavykeeper()
+       // {
+
+      //  }