From c092867bb322e35ee2de0ceace804e4032000267 Mon Sep 17 00:00:00 2001
From: Harshil Goel <harshil@hypermode.com>
Date: Mon, 11 Nov 2024 04:27:54 +0530
Subject: [PATCH] added something

---
 countmin.go | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/countmin.go b/countmin.go
index 2af125a..68a5858 100644
--- a/countmin.go
+++ b/countmin.go
@@ -93,6 +93,17 @@ func (c *CountMinSketch) Add(data []byte) *CountMinSketch {
 	return c
 }
 
+func (c *CountMinSketch) Set(data []byte, count uint64) *CountMinSketch {
+	lower, upper := hashKernel(data, c.hash)
+
+	// Increment count in each row.
+	for i := uint(0); i < c.depth; i++ {
+		c.matrix[i][(uint(lower)+uint(upper)*i)%c.width] = count
+	}
+
+	return c
+}
+
 // Count returns the approximate count for the specified item, correct within
 // epsilon * total count with a probability of delta.
 func (c *CountMinSketch) Count(data []byte) uint64 {