Create Solution.js

Author: 07subhadip

solution/1300-1399/1381.Design a Stack With Increment Operation/Solution.js

@@ -0,0 +1,52 @@
+/**
+ * @param {number} maxSize
+ */
+var CustomStack = function(maxSize) {
+    this.stk = Array(maxSize).fill(0);
+    this.add = Array(maxSize).fill(0);
+    this.i = 0;
+};
+
+/** 
+ * @param {number} x
+ * @return {void}
+ */
+CustomStack.prototype.push = function(x) {
+    if (this.i < this.stk.length) {
+        this.stk[this.i++] = x;
+    }
+};
+
+/**
+ * @return {number}
+ */
+CustomStack.prototype.pop = function() {
+    if (this.i <= 0) {
+        return -1;
+    }
+    const ans = this.stk[--this.i] + this.add[this.i];
+    if (this.i > 0) {
+        this.add[this.i - 1] += this.add[this.i];
+    }
+    this.add[this.i] = 0;
+    return ans;
+};
+
+/** 
+ * @param {number} k 
+ * @param {number} val
+ * @return {void}
+ */
+CustomStack.prototype.increment = function(k, val) {
+    if (this.i > 0) {
+        this.add[Math.min(this.i, k) - 1] += val;
+    }
+};
+
+/** 
+ * Your CustomStack object will be instantiated and called as such:
+ * var obj = new CustomStack(maxSize)
+ * obj.push(x)
+ * var param_2 = obj.pop()
+ * obj.increment(k,val)
+ */