โจทย์
องค์ประกอบที่น้อยกว่าถัดไปคือองค์ประกอบที่เป็นองค์ประกอบแรกที่น้อยกว่าหลังจากมัน มาดูตัวอย่างกัน
arr = [1, 2, 3, 5, 4]
องค์ประกอบที่น้อยกว่าถัดไปสำหรับ 5 คือ 4 และองค์ประกอบที่น้อยกว่าถัดไปสำหรับองค์ประกอบ 1, 2, 3 คือ -1 เนื่องจากไม่มีองค์ประกอบที่น้อยกว่าหลังจากพวกมัน
อัลกอริธึม
- กำหนดอาร์เรย์ด้วยตัวเลขสุ่ม
- กำหนดสแต็ก (stack)
- เพิ่มองค์ประกอบแรกเข้าไปในสแต็ก
- วนลูปผ่านองค์ประกอบของอาร์เรย์
- ถ้าสแต็กว่าง ให้เพิ่มองค์ประกอบปัจจุบันเข้าไปในสแต็ก
- ขณะที่องค์ประกอบปัจจุบันน้อยกว่าองค์ประกอบบนสุดของสแต็ก
- พิมพ์องค์ประกอบบนสุดพร้อมกับองค์ประกอบที่น้อยกว่าถัดไปเป็นองค์ประกอบปัจจุบัน
- นำองค์ประกอบบนสุดออก (pop)
- เพิ่มองค์ประกอบเข้าไปในสแต็ก
- ขณะที่สแต็กไม่ว่าง
- พิมพ์องค์ประกอบพร้อมกับองค์ประกอบที่น้อยกว่าถัดไปเป็น -1
การนำไปใช้งาน
ต่อไปนี้เป็นการนำไปใช้งานของอัลกอริธึมข้างต้นใน C++
#include <bits/stdc++.h>
using namespace std;
void nextSmallerElements(int arr[], int n) {
stack<int> s;
s.push(arr[0]);
for (int i = 1; i < n; i++) {
if (s.empty()) {
s.push(arr[i]);
continue;
}
while (!s.empty() && s.top() > arr[i]) {
cout << s.top() << " -> " << arr[i] << endl;
s.pop();
}
s.push(arr[i]);
}
while (!s.empty()) {
cout << s.top() << " -> " << -1 << endl;
s.pop();
}
}
int main() {
int arr[] = { 5, 4, 3, 2, 1 };
int n = 5;
nextSmallerElements(arr, n);
return 0;
}
ผลลัพธ์
หากคุณรันโค้ดข้างต้น คุณจะได้ผลลัพธ์ดังนี้
1 -> 2 2 -> 3 3 -> 4 4 -> 5 5 -> -1
