องค์ประกอบที่มากกว่าถัดไปในลำดับเดียวกับอินพุตใน C++

โจทย์

องค์ประกอบที่มากกว่าถัดไปคือองค์ประกอบที่เป็นองค์ประกอบแรกที่มากกว่าหลังจากมัน มาดูตัวอย่างกัน

arr = [4, 5, 3, 2, 1]

องค์ประกอบที่มากกว่าถัดไปสำหรับ 4 คือ 5 และองค์ประกอบที่มากกว่าถัดไปสำหรับองค์ประกอบ 3, 2, 1 คือ -1 เนื่องจากไม่มีองค์ประกอบที่มากกว่าหลังจากพวกมัน

อัลกอริธึม

• กำหนดอาร์เรย์ด้วยตัวเลขสุ่ม
• กำหนดสแต็ก (stack) และอาร์เรย์
• วนลูปจากท้ายอาร์เรย์
  • นำองค์ประกอบออกจากสแต็กจนกว่าจะว่างและองค์ประกอบบนสุดน้อยกว่าหรือเท่ากับองค์ประกอบปัจจุบัน
  • ถ้าสแต็กว่าง แสดงว่าไม่มีองค์ประกอบที่มากกว่าถัดไป เพิ่ม -1 เข้าไปในอาร์เรย์ผลลัพธ์
  • ถ้าสแต็กไม่ว่าง องค์ประกอบบนสุดคือองค์ประกอบที่มากกว่าถัดไปสำหรับองค์ประกอบปัจจุบัน เพิ่มเข้าไปในอาร์เรย์ผลลัพธ์
  • เพิ่มองค์ประกอบปัจจุบันเข้าไปในสแต็ก
• วนลูปผ่านอาร์เรย์ผลลัพธ์และพิมพ์แต่ละองค์ประกอบพร้อมกับองค์ประกอบที่มากกว่าถัดไป

การนำไปใช้งาน

ต่อไปนี้เป็นการนำไปใช้งานของอัลกอริธึมข้างต้นใน C++

#include <bits/stdc++.h> 
using namespace std; 

void nextGreaterElements(int arr[], int n) { 
    stack<int> s; 
    int result[n]; 
    for (int i = n - 1; i >= 0; i--) { 
        while (!s.empty() && s.top() <= arr[i]) { 
            s.pop(); 
        } 
        if (s.empty()) { 
            result[i] = -1; 
        }else { 
            result[i] = s.top(); 
        } 
        s.push(arr[i]); 
    } 
    for (int i = 0; i < n; i++) { 
        cout << arr[i] << " -> " << result[i] << endl; 
    } 
} 

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