โจทย์
องค์ประกอบที่มากกว่าถัดไปคือองค์ประกอบที่เป็นองค์ประกอบแรกที่มากกว่าหลังจากมัน มาดูตัวอย่างกัน
arr = [4, 5, 3, 2, 1]
องค์ประกอบที่มากกว่าถัดไปสำหรับ 4 คือ 5 และองค์ประกอบที่มากกว่าถัดไปสำหรับองค์ประกอบ 3, 2, 1 คือ -1 เนื่องจากไม่มีองค์ประกอบที่มากกว่าหลังจากพวกมัน
อัลกอริธึม
- กำหนดอาร์เรย์ด้วยตัวเลขสุ่ม
- กำหนดสแต็ก (stack)
- เพิ่มองค์ประกอบแรกเข้าไปในสแต็ก
- วนลูปผ่านองค์ประกอบของอาร์เรย์
- ถ้าสแต็กว่าง ให้เพิ่มองค์ประกอบปัจจุบันเข้าไปในสแต็ก
- ขณะที่องค์ประกอบปัจจุบันมากกว่าองค์ประกอบบนสุดของสแต็ก
- พิมพ์องค์ประกอบบนสุดพร้อมกับองค์ประกอบที่มากกว่าถัดไปเป็นองค์ประกอบปัจจุบัน
- นำองค์ประกอบบนสุดออก (pop)
- เพิ่มองค์ประกอบเข้าไปในสแต็ก
- ขณะที่สแต็กไม่ว่าง
- พิมพ์องค์ประกอบพร้อมกับองค์ประกอบที่มากกว่าถัดไปเป็น -1
การนำไปใช้งาน
ต่อไปนี้เป็นการนำไปใช้งานของอัลกอริธึมข้างต้นใน C++
#include <bits/stdc++.h>
using namespace std;
void nextGreaterElements(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[] = { 1, 2, 3, 4, 5 };
int n = 5;
nextGreaterElements(arr, n);
return 0;
}
ผลลัพธ์
หากคุณรันโค้ดข้างต้น คุณจะได้ผลลัพธ์ดังนี้
1 -> 2 2 -> 3 3 -> 4 4 -> 5 5 -> -1
