আমরা একটি স্ট্যাক তৈরি করতে চাই যাতে সেটি সর্বোচ্চ উপাদান সংরক্ষণ করতে পারে এবং O(1) সময়ে সেটি পাওয়া যায়। শর্ত হলো এটি O(1) অতিরিক্ত স্পেস ব্যবহার করবে।
আমরা একটি কাস্টম স্ট্যাক তৈরি করব যা পुश (push), পপ (pop) বা পিক (peek) অপারেশন করার সময় সর্বোচ্চ মান আপডেট করবে। পিক অপারেশনের জন্য স্ট্যাকের টপ (top) এবং সর্বোচ্চ উপাদানের মধ্যে বড়টি রিটার্ন করবে, পপ অপারেশনের সময় যদি টপ উপাদানটি বড় হয়, তবে সেটি প্রিন্ট করে সর্বোচ্চ মান আপডেট করবে 2*max - top_element হিসেবে। অন্যথায় টপ উপাদানটি রিটার্ন করবে। পশ অপারেশনের সময় সর্বোচ্চ মান আপডেট হবে x (ইনসার্ট করার ডেটা), 2*x - max হিসেবে।
উদাহরণ
#include <iostream> #include <stack> using namespace std; class CustomStack { stack<int> stk; int stack_max; public: void getMax() { if (stk.empty()) cout << "Stack is empty"<<endl; else cout << "Maximum Element in the stack is: "<< stack_max <<endl; } void peek() { if (stk.empty()) { cout << "Stack is empty "; return; } int top = stk.top(); // Top element. cout << "Top Most Element is: "<<endl; (top > stack_max) ? cout << stack_max : cout << top; } void pop() { if (stk.empty()) { cout << "Stack is empty"<<endl; return; } cout << "Top Most Element Removed: "; int top = stk.top(); stk.pop(); if (top > stack_max) { cout << stack_max <<endl; stack_max = 2 * stack_max - top; } else cout << top <<endl; } void push(int element) { if (stk.empty()) { stack_max = element; stk.push(element); cout << "Element Inserted: " << element <<endl; return; } if (element > stack_max) { stk.push(2 * element - stack_max); stack_max = element; } else stk.push(element); cout << "Element Inserted: " << element <<endl; } }; int main() { CustomStack stk; stk.push(4); stk.push(6); stk.getMax(); stk.push(8); stk.push(20); stk.getMax(); stk.pop(); stk.getMax(); stk.pop(); stk.peek(); }আউটপুট
Element Inserted: 4 Element Inserted: 6 Maximum Element in the stack is: 6 Element Inserted: 8 Element Inserted: 20 Maximum Element in the stack is: 20 Top Most Element Removed: 20 Maximum Element in the stack is: 8 Top Most Element Removed: 8 Top Most Element is: 6
